fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. int t;
  6. cin>>t;
  7. while(t--)
  8. {
  9. int n;
  10. cin>>n;
  11. int a[n];
  12. for(int i=0;i<n;i++)
  13. {
  14. cin>>a[i];
  15. }
  16. int s=0,st=0,s1=0,c=0;
  17. for(int i=0;i<n;i++)
  18. {
  19. if(s<a[i] && i>0)
  20. {
  21. st=-s+i*a[i];
  22. s1=-1;c=-1;
  23. }
  24. else if(s1<a[i] && c>0)
  25. {
  26. st=st-s1+c*a[i];
  27. }
  28. if(s1==-1)
  29. s1=0;
  30. else
  31. s1=s1+a[i];
  32. s=s+a[i];c++;
  33.  
  34. }
  35.  
  36. cout<<st<<endl;
  37. }
  38. return 0;
  39. }
Success #stdin #stdout 0.01s 5404KB
stdin
3
3
5 3 2
3
1 2 100
4
1 3 1 2
stdout
0
197
3