fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define int long long
  5. #define endl '\n'
  6. #define rep(i , l , n) for(int i = l ; i <= n ; i++)
  7.  
  8. const int N = 1e6;
  9.  
  10.  
  11. int n , k , ans = 0;
  12. int a[N] , pre[N];
  13. unordered_map < int , int > d;
  14.  
  15. signed main()
  16. {
  17.     ios::sync_with_stdio(0);
  18.     cin.tie(0);
  19.  
  20.     cin >> n >> k;
  21.  
  22.     rep(i , 1 , n)
  23.     {
  24.         cin >> a[i];
  25.         pre[i] = pre[i - 1] + a[i];
  26.     }
  27.  
  28.     d[0] = 1; 
  29.  
  30.     rep(i , 1 , n)
  31.     {
  32.         int adr = pre[i] - k * i;
  33.  
  34.         ans += d[adr];
  35.  
  36.         d[adr]++;
  37.     }
  38.  
  39.     cout << ans << endl;
  40.  
  41.     return 0;
  42. }
  43.  
Success #stdin #stdout 0.01s 5616KB
comments (?)
stdin
copy 
5 2
1 3 -2 3 -5
stdout
copy 
1
https://ideone.com/Ukwz4N
language:
C++14 (gcc 8.3)
created:
7 hours 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!