fork download
  1. #include<iostream>
  2. #include<bits/stdc++.h>
  3. using namespace std;
  4. #define ll long long
  5. long long a[1000000];
  6. long long freq[1000000];
  7. int main() {
  8. int queries;
  9. cin>>queries;
  10. while(queries--)
  11. {long long int n;
  12. cin>>n;
  13. ll t;
  14. ll ans=0;
  15. ll sum=0;
  16. memset(freq,0,sizeof(freq));
  17. freq[0]=1;
  18. for(ll int i=0;i<n;i++)
  19. {cin>>a[i];
  20. sum=sum+a[i];
  21. t=sum;
  22. t=(t+n)%n;
  23. freq[t]++;
  24. }
  25. for(ll int i=0;i<n /*and( freq[i]>=2)*/;i++)
  26. {ll int m=freq[i];
  27. ans +=(m)*(m-1)/2;
  28. }
  29. cout<<ans<<endl;
  30. }
  31. return 0;
  32. }
Success #stdin #stdout 0s 4732KB
stdin
Standard input is empty
stdout
Standard output is empty