Advertisement
Korotkodul

E

Dec 10th, 2022
680
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.74 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int, int>
  11. #define pb(x) push_back(x)
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v) {
  17.     for (auto x : v) cout << x << ' ';
  18.     cout << "\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v) {
  22.     for (auto x : v) cout << x << ' ';
  23.     cout << "\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v) {
  28.     for (auto x : v) cv(x);
  29.     cout << "\n";
  30. }
  31.  
  32. void cvb(vector <bool> v) {
  33.     for (bool x : v) cout << x << ' ';
  34.     cout << "\n";
  35. }
  36.  
  37. void cvs(vector <string>  v) {
  38.     for (auto a : v) {
  39.         cout << a << "\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a) {
  44.     for (auto p : a) {
  45.         cout << p.first << ' ' << p.second << "\n";
  46.     }
  47.     cout << "\n";
  48. }
  49.  
  50. bool sh = 0;
  51.  
  52.  
  53. int main() {
  54.     ios::sync_with_stdio(0);
  55.     cin.tie(0);
  56.     cout.tie(0);
  57.     ll n, m; cin >> n  >> m;
  58.     vector <ll> a(n), cnt(m, 0);
  59.     for (int i = 0; i < n; ++i) {
  60.         cin >> a[i];
  61.         a[i] %= m;
  62.     }
  63.     ll s = 0, ans = 0;
  64.     if (sh) {
  65.         cout << "n m = " << n << ' ' << m << "\n";
  66.         cout << "a\n";
  67.         cvl(a);
  68.         cout << "cnt\n";
  69.         cvl(cnt);
  70.         cout << "\n\n";
  71.     }
  72.     for (int i = 0; i < n; ++i) {
  73.         if (sh) cout << "i = " << i << "\n";
  74.         s += a[i];
  75.         s %= m;
  76.         if (sh) cout << "s = " << s << "\n";
  77.         ans += cnt[s];
  78.         cnt[s]++;
  79.         if (sh) {
  80.             cout << "ans = " << ans << "\n";
  81.             cout << "cnt\n";
  82.             cvl(cnt);
  83.             cout << "\n\n";
  84.         }
  85.  
  86.     }
  87.     ans += cnt[0];
  88.     cout << ans;
  89. }
  90.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement