fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. void executeTime() {
  5. cerr << "Time Taken: " << (float)clock() / CLOCKS_PER_SEC << " secs";
  6. }
  7.  
  8. int main() {
  9.  
  10. int n; cin >> n;
  11.  
  12. int x, y; cin >> x >> y;
  13.  
  14. vector<int> v(n);
  15. for (auto &x : v) {
  16. cin >> x;
  17. }
  18.  
  19.  
  20. unordered_map<int, int> mp;
  21. mp[0]++;
  22.  
  23. int curr = 0, ans = 0;
  24. for (int j = 0; j < n; j++) {
  25. curr += (v[j] == x ? 1 : -1);
  26. ans += mp[curr];
  27.  
  28. mp[curr]++;
  29. }
  30.  
  31. cout << ans << endl;
  32.  
  33. executeTime();
  34. return 0;
  35. }
Success #stdin #stdout #stderr 0s 5312KB
stdin
9
5 4
5 4 5 4	5 5 4 4 5

stdout
16
stderr
Time Taken: 0.004087 secs