fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main()
  5. {
  6.     long long t,n;
  7.     int c=0;
  8.     cin>>t;
  9.     for(long long i=0;i<t;i++)
  10.     {
  11.         cin>>n;
  12.         int a[n];
  13.         for(long long j=0;j<n;j++)
  14.  
  15.         {
  16.             cin>>a[j];
  17.         }
  18.         int ig;
  19.         if(c==t-1)
  20.         {
  21.             cin>>ig;
  22.         }
  23.  
  24.         int m=0,l=0;
  25.         long long int w=0;
  26.         while(m!=n && l!=n)
  27.         {
  28.             if(a[m]>0 && a[l]<0)
  29.             {
  30.                 if(a[m]>=abs(a[l]))
  31.                 {
  32.                     a[m]=a[m]-abs(a[l]);
  33.                     w += abs(l-m)*abs(a[l]);
  34.                     a[l]=0;
  35.                     l++;
  36.                 }
  37.                 else
  38.                 {
  39.                     a[l]=a[l]+a[m];
  40.                      w += abs(l-m)*(a[m]);
  41.                     a[m]=0;
  42.                     m++;
  43.                 }
  44.  
  45.             }
  46.             if(a[m]<=0)
  47.             {
  48.                 m++;
  49.             }
  50.             if(a[l]>=0)
  51.             {
  52.                 l++;
  53.             }
  54.  
  55.  
  56.         }
  57.         cout<<w<<" ";
  58.         c++;
  59.     }
  60.  
  61.  
  62.  
  63.     return 0;
  64. }
Success #stdin #stdout 0.01s 5424KB
comments (?)
stdin
copy 
1
5
5 -4 1 -3 1
stdout
copy 
9
https://ideone.com/BE0MD7
language:
C++14 (gcc 8.3)
created:
1 year ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!