fork download 
  1. #include<bits/stdc++.h>
  2. #define TIME (1.0* clock()/CLOCKS_PER_SEC)
  3. #define pb push_back
  4. #define eb emplace_back
  5. #define id1 (id<<1)
  6. #define id2 (id<<1)|1
  7. #define ll long long
  8. #define ii pair<int,int>
  9. #define vi vector<int>
  10. #define vii vector<pair<int,int>> 
  11. #define vl vector<long long>
  12. #define vll vector <pair<ll,ll>>
  13. #define li pair<long long,int>
  14. #define vil vector <pair<int,ll>>
  15. #define db double
  16. #define ff first
  17. #define ss second
  18. #define lb lower_bound
  19. #define ub upper_bound
  20. #define FOR(i, a, b) for (int i = (a); i <=(b); i++)
  21. #define F0R(i, a) FOR(i, 0, a-1)
  22. #define ROF(i, a, b) for (int i = (b); i >= (a); i--)
  23. #define R0F(i, a) ROF(i, 0, a-1)
  24. #define rep(a) F0R(_, a)
  25. #define each(a, x) for (auto &a : x)
  26. #define ALL(x) (x).begin(),(x).end()
  27. #define pq priority_queue <li, vector <li>, greater <li>> 
  28. using namespace std;
  29. const int maxn=1e6;
  30. //const int MOD=1e9+7;
  31. //const int MOD=998244353;
  32. //const int dx[4]{1, 0, -1, 0}, dy[4]{0, 1, 0, -1};
  33. #define int long long
  34.  
  35. int n,d;
  36. int a[maxn];
  37.  
  38. void solve(){
  39.    cin>>n>>d;
  40.  
  41.    unordered_map<int,int> cnt;
  42.    int res = 0;
  43.    int pref = 0; 
  44.    cnt[pref] = 1;
  45.    for(int i=1;i<=n;i++){
  46.    	 int x;
  47.    	 cin>>x;
  48.    	 pref = (pref + x) % d;
  49.    	 if(pref < 0) pref += d;
  50.    	 res += cnt[pref];
  51.    	 cnt[pref]++;
  52.    }
  53.    cout<<res<<"\n";
  54.    return;
  55. }
  56. signed main(){
  57.   ios_base::sync_with_stdio(false);
  58.   cin.tie(NULL);cout.tie(NULL);
  59.   if (fopen("TASK.INP", "r")){
  60.       freopen("TASK.INP", "r", stdin);
  61.       freopen("TASK.OUT", "w", stdout);
  62.   }
  63.   int ntest;
  64.   ntest=1;
  65.   cin>>ntest;
  66.  
  67.   for(int i=1;i<=ntest;i++) solve();
  68.   cerr<<"\n"<<"Time elapsed "<<TIME<<"s.\n";
  69.   return 0;
  70. }
  71.  
Success #stdin #stdout #stderr 0.01s 5320KB
comments (?)
stdin 
5 4
1 3 -2 3 -5
stdout 
10
10
10
10
10
stderr 
Time elapsed 0.004913s.
https://ideone.com/N3gqCA
language:
C++ (gcc 8.3)
created:
6 days 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!