Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <bits/stdc++.h>
- using namespace std;
- map <int, int> cnt;
- int n, x, y;
- int32_t main()
- {
- cin >> n >> x >> y;
- int a [n];
- cin >> a [0];
- a [0] = a [0] % x;
- for (int i = 1; i < n; i++)
- {
- cin >> a [i];
- a [i] = (a [i] + a [i - 1]) % x;
- }
- int ans = 0;
- for (int i = 0; i < n; i++)
- {
- if (a [i] == y)
- {
- ans += cnt [0] + 1;
- cnt [a [i]]++;
- continue;
- }
- if (a [i] > y)
- {
- ans += cnt [a [i] - y];
- cnt [a [i]]++;
- continue;
- }
- if (a [i] < y)
- {
- ans += cnt [a [i] + x - y];
- cnt [a [i]]++;
- continue;
- }
- }
- cout << ans;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement