fork download
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. typedef long long ll;
  6.  
  7. int main()
  8. {
  9. ios_base::sync_with_stdio( false ), cin.tie( nullptr ), cout.tie( nullptr );
  10.  
  11. int N, M, A; cin >> N >> M; ll sum = 0, pairs = 0; map< int, int > count;
  12.  
  13. for( int i = 0; i < N; i++ )
  14. cin >> A, sum += A, sum %= M, count[ sum ]++;
  15.  
  16. for( auto p: count )
  17. {
  18. ll k = p.second;
  19.  
  20. if ( p.first == 0 )
  21. k *= k + 1;
  22. else
  23. k *= k - 1;
  24.  
  25. k >>= 1, pairs += k;
  26. }
  27.  
  28. cout << pairs;
  29. }
Success #stdin #stdout 0s 4488KB
stdin
13 17
29 7 5 7 9 51 7 13 8 55 42 9 81
stdout
6