洛谷 P1599 结算日

P1599 结算日

题目背景

借债还债

题目描述

“不放债不借债”，贝西多么希望自己可以遵循这个忠告。她已经和她的N（1 <= N <= 100,000）个朋友有了债务关系，或者借债了，或者放债了。她的N个朋友依次标号为1..N。 结算日终于来临了。她知道，朋友欠她的钱比她欠朋友的钱多。她的朋友们分布在一条直线上，第i头奶牛站的位置距离谷仓i米。贝西打算沿着这条直线行走，从欠她钱的奶牛手里收钱回来，并且还钱给她欠钱的奶牛。 当她沿直线移动的时候，她可以要求任何欠她钱的奶牛还全部的钱。当她有足够的钱可以还清她的某个债，就可以把钱给对应的奶牛还清她的债。奶牛i欠贝西D_i元（-1,000 <= D_i <=1,000; D_i <> 0），负数表示贝西欠奶牛i钱。 贝西从谷仓出发，位置为0，初始贝西没有钱。贝西收回她的所有借债，并且还清她的欠债所需行走的最短距离是多少？注意：她必须在最后一头奶牛所在的位置，完成她的行走。

输入输出格式

输入格式：

 

行1：一个整数：N 行2..N+1：第i+1行包含一个整数：Di

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

5
100
-200
250
-200
200

输出样例#1： 复制

9

说明

输入解释：

3头奶牛欠贝西钱；她欠2头奶牛钱。当她完成结算，她将有150元。

输出解释：

谷仓  100  -200  250 -200  200
 |     |     |    |    |    |
 ***>**+**>*****>**+
                   *            < 贝西有 350元
             -**<***
             *                  < 贝西有 150元
             ***>****>****>**+
                             *  < 贝西有 350
                       -**<***
                       *     
                       ***>***  < 贝西结束她的行走，有 150元
思路：模拟

#include<iostream>
using namespace std;
int n,x,sum,ans,l;
bool flag;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>x;sum+=x;ans++;
        if((sum>=0)&&flag){flag=0;ans+=(i-l)*2;}
        if((sum<0)&&!flag){flag=1;l=i;}
    }
    cout<<ans;
}