Submission #2368661


Source Code Expand

#include <bits/stdc++.h>
#define rep(i,n) for(int i=0;i<(n);i++)
#define loop(i,x,n) for(int i=(x);i<(n);i++)
#define all(v) (v).begin(),(v).end()
#define int long long
using namespace std;
const int MOD=1e9+7;
const int INF=1e10;

signed main(){
  int n;
  cin>>n;
  vector<int> a(n);
  rep(i,n)cin>>a[i];
  int sum=0;

  sum=a[0];
  int ans1=0,tmp;
  rep(i,n-1){
    if(sum==0&&i==0){
      sum=1;
      ans1++;
    }
    if(sum>0&&sum+a[i+1]>0){
      sum+=a[i+1];
      tmp=abs(-1-sum);
      sum-=tmp;
      ans1+=tmp;
      if(sum==0){
        sum--;
        ans1++;
      }
    }else if(sum<0&&sum+a[i+1]<0){
      sum+=a[i+1];
      tmp=1-sum;
      sum+=tmp;
      ans1+=tmp;
      if(sum==0){
        sum++;
        ans1++;
      }else{
        sum+=a[i+1];
      }
    }
  }

  int ans2=0;
  sum=a[0];
  rep(i,n-1){
    if(sum==0&&i==0){
      sum=-1;
      ans2++;
    }
    if(sum>0&&sum+a[i+1]>0){
      sum+=a[i+1];
      tmp=abs(-1-sum);
      sum-=tmp;
      ans2+=tmp;
      if(sum==0){
        sum--;
        ans2++;
      }
    }else if(sum<0&&sum+a[i+1]<0){
      sum+=a[i+1];
      tmp=1-sum;
      sum+=tmp;
      ans2+=tmp;
      if(sum==0){
        sum++;
        ans2++;
      }
    }else{
      sum+=a[i+1];
    }
  }
  cout<<min(ans1,ans2)<<endl;

  return 0;
}

Submission Info

Submission Time
Task C - Sequence
User fuu32
Language C++14 (GCC 5.4.1)
Score 0
Code Size 1369 Byte
Status WA
Exec Time 46 ms
Memory 1024 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 300
Status
AC × 2
WA × 1
AC × 9
WA × 9
Set Name Test Cases
Sample 00-00.txt, 00-01.txt, 00-02.txt
All 00-00.txt, 00-01.txt, 00-02.txt, 01-00.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt
Case Name Status Exec Time Memory
00-00.txt AC 1 ms 256 KB
00-01.txt AC 1 ms 256 KB
00-02.txt WA 1 ms 256 KB
01-00.txt WA 44 ms 1024 KB
01-01.txt WA 44 ms 1024 KB
01-02.txt WA 43 ms 1024 KB
01-03.txt WA 41 ms 1024 KB
01-04.txt WA 39 ms 896 KB
01-05.txt AC 35 ms 896 KB
01-06.txt WA 42 ms 1024 KB
01-07.txt AC 38 ms 896 KB
01-08.txt AC 46 ms 1024 KB
01-09.txt AC 39 ms 896 KB
01-10.txt WA 32 ms 768 KB
01-11.txt AC 43 ms 1024 KB
01-12.txt AC 44 ms 1024 KB
01-13.txt WA 39 ms 896 KB
01-14.txt AC 44 ms 1024 KB