16、公交系统

【问题描述】

        城市公交系统有一个记录仪，用于记录每个站点的乘客人数的变化情况，例如：x表示到站前公交车上的乘客人数，y表示离站时公交车上的乘客人数，则该记录仪记录的该站的数字为y-x。

        对于一辆公交车和n个车站，a₁,a₂,...,a_n为该公交车在各站的记录数据。

        假定w为该公交车可容纳的最大乘客人数，编程求出在第一站停靠之前公交车上人数的可能数据有多少种？

【输入形式】

        第一行包含两个数据n和w(1<=n<=1000, 1<=w<=10⁹)，分别表示车站的数目和公交车可容纳的最大乘客人数。

        第二行包含一个序列a₁,a₂,...,a_n，表示记录仪记录的各站的数据。

【输出形式】

        输出一个整数，表示公交车在第一站停靠之前可能的乘客人数数据的个数，如果没有，则输出0。

【样例输入1】

3 5
2 1 -3

【样例输出1】

3

【样例输入2】

2 4
-1 1

【样例输出2】

4

【样例输入3】

4 10
2 4 1 2

【样例输出3】

2

【样例说明】

在第一个样例中，乘客数可能有0、1、2，共3种情况
在第二个样例中，乘客数可能有1、2、3、4，共4种情况
在第三个样例种，乘客数可能为0或1，共2种情况

【my code】

#include <iostream>
using namespace std;
int main()
{
 int n,w,s[100],i,k=1,count1=0,count2=0,count[100]={0},t=0;
 cin>>n>>w;
 for(i=0;i<n;i++){
  cin>>s[i];
 }
 for(i=0;i<n;i++){
  count1+=s[i];
  count[i]=count1;
  if(count1>=-w&&count1<=w){
   if(count1<0){
    count2=-count1;
   }
  }
  else{
   k=0;break;
  }
 }
 for(i=1;i<n;i++){
  if(count[i]>count[0]){
   t=count[0];
   count[0]=count[i];
   count[i]=t;
  }
 }
  if(k!=0){
   if(count2>0){
    cout<<w-count2+1<<endl;
    }
   else{
    cout<<w-count[0]+1<<endl; 
   }
   }
  else{
   cout<<k<<endl;
  }
 return 0;
}