走格子

有编号1-n的n个格子，机器人从1号格子顺序向后走，一直走到n号格子，并需要从n号格子走出去。机器人有一个初始能量，每个格子对应一个整数A[i]，表示这个格子的能量值。如果A[i] > 0，机器人走到这个格子能够获取A[i]个能量，如果A[i] < 0，走到这个格子需要消耗相应的能量，如果机器人的能量 < 0，就无法继续前进了。问机器人最少需要有多少初始能量，才能完成整个旅程。

例如：n = 5。{1，-2，-1，3，4} 最少需要2个初始能量，才能从1号走到5号格子。途中的能量变化如下3 1 0 3 7。

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

long long int fun(vector<long long int> &input, int N)
{
	long long int result = 0;
	long long int sum = 0;
	for (int i = 0; i < N; i++)
	{
		sum += input[i];
		if (sum < 0)
		{
			result += -sum;
			sum = 0;
		}
	}
	
	return result;
}


int main() {
	// your code goes here
	int N;
	scanf("%d", &N);
	
	vector<long long int> input;
	long long int a;
	for (int i = 0; i < N; i++)
	{
	    scanf("%lld", &a);
	    input.push_back(a);
	}
	
	cout << fun(input, N) << endl;
	return 0;
}