HNU程序设计训练——公交系统

【问题描述】

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

        对于一辆公交车和n个车站，a1,a2,...,an为该公交车在各站的记录数据。

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

【输入形式】

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

        第二行包含一个序列a1,a2,...,an，表示记录仪记录的各站的数据。

【输出形式】

        输出一个整数，表示公交车在第一站停靠之前可能的乘客人数数据的个数，如果没有，则输出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种情况

#include <iostream>
#include <vector>
using namespace std;

int main()
{
	int n, w;
	vector<int> a;
	cin >> n >> w;
	for (int i = 0; i < n; i++)
	{
		int tmp;
		cin >> tmp;
		a.push_back(tmp);
	}
	int head = 0, tail = w,curr=0;
	for (int i = 0; i < n; i++)
	{
		curr += a[i];
		if (curr < -head&& curr < 0)
			head = -curr;
		else if (w - curr >= 0 && w - curr < tail)
			tail = w - curr;
		else if (w - curr < 0)
		{
			tail = 0;
			head = 1;
			break;
		}
	}
	if (head > tail)
		cout << 0;
	else
		cout << tail - head + 1;
	return 0;
}

curr表示公交车当前人数，逐步缩小head（可能的最小人数）与tail（可能的最大人数）的范围