小J还债题解

小j还债

描述
随着小J的商业帝国越来越大，他的债务关系也越来越复杂。 现在他与N个朋友有个债务关系,1<=N<=100000 现在已知小J朋友欠她的钱比她欠朋友的钱多。 她的朋友们分布在一条直线上，所有点相隔的距离均为1米 小J打算从自己家出发沿着这条直线行走，从欠她钱的朋友手里收钱回来，并且还钱给她欠钱的朋友。
当她沿直线移动的时候，她可以要求任何欠她钱的朋友还全部的钱。 当她有足够的钱可以还清她的某个债，就可以把钱给对应的朋友还清她的债。 朋友i欠小J为D_i元（-1,000 <= D_i <=1,000; D_i <> 0），负数表示小J欠朋友D_i钱。
小J从谷仓出发，位置为0，初始小J没有钱。 小J收回她的所有借债，并且还清她的欠债所需行走的最短距离是多少？ 注意：她必须在最后一头朋友所在的位置，完成她的行走。

输入
行1：一个整数：N 行2…N+1：第i+1行包含一个整数：Di

输出
行1：一个整数，贝西收回借债并且还清欠债，所需要行走的最短距离（单位为米）

样例
输入
5
100
-200
250
-200
200
输出
9

完整代码

#include <bits/stdc++.h>
using namespace std;
int n,ans,sum,kao;//kao表示欠没欠钱
int main(){
	cin>>n;
	for(int i=1,a;i<=n;i++){
		cin>>a;
		sum+=a;
		if(sum>=0&&kao)ans+=(i-kao)*2,kao=0;
		if(sum<0&&!kao)kao=i;
	}
	cout<<ans+n;
	return 0;
}