fork download
  1. #include <iostream>
  2. #include <map>
  3. using namespace std;
  4.  
  5. int main() {
  6.  
  7. long long n;
  8. cin>>n;
  9.  
  10. long long a[n];
  11. for(long long i=0;i<n;i++)
  12. {
  13. cin>>a[i];
  14. }
  15.  
  16. long long sum=0;
  17. long long count1=0;
  18. map<long long,long long> Map;
  19. Map[0]=1;
  20. for(long long i=0;i<n;i++)
  21. {
  22. sum=((sum+a[i])%n+n)%n;
  23. if(Map.count(sum) > 0)
  24. {
  25. count1+=Map[sum];
  26. }
  27. if(Map.count(sum) <= 0)
  28. {
  29. Map[sum]=1;
  30. }
  31. else
  32. {
  33. Map[sum]++;
  34. }
  35. }
  36.  
  37. cout<<count1<<endl;
  38.  
  39. return 0;
  40. }
Success #stdin #stdout 0s 5272KB
stdin
100
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
stdout
1