Advertisement
Korotkodul

ege_27_dp_WA

Feb 16th, 2023 (edited)
956
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.47 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. #include <fstream>
  51.  
  52.  
  53. int main() {
  54.     string data;
  55.     ifstream file("27.txt");
  56.     getline(file, data);
  57.     int N = stoi(data);
  58.     vector <int> a;
  59.     int ans = 0;
  60.     for (int k = 0; k < N; ++k) {
  61.         getline(file, data);
  62.         int x = stoi(data) % 1111;
  63.         a.pb(x);
  64.         if (x == 0) {
  65.             ans++;
  66.         }
  67.     }
  68.     /*int Si = 0, Sj;
  69.     cout << "N = " << N << "\n";
  70.     for (int i = 0; i < N; ++i) {
  71.         Si += a[i];
  72.         Si %= 1111;
  73.         Sj = 0;
  74.         for (int j = 0; j < i; ++j) {
  75.             Sj += a[j];
  76.             Sj %= 1111;
  77.             if (Si == Sj) {
  78.                 cout << i << ' ' << j << "\n";
  79.                 ans++;
  80.             }
  81.         }
  82.     }
  83.     cout << ans;*/
  84.     cout << "N = " << N << "\n";
  85.     vector <int> pf(N);
  86.     pf[0] = a[0];
  87.     for (int i = 1; i < N; ++i) {
  88.         pf[i] = pf[i - 1] + a[i];
  89.         pf[i] %= 1111;
  90.     }
  91.     vector <vector <int> > cnt(1111, vector <int> (N, 0));
  92.     for (int i = 0; i < N; ++i) {
  93.         cnt[pf[i]][i]++;
  94.     }
  95.     cnt[a[0]][0]=1;
  96.     for (int i = 1; i < N; ++i) {
  97.         for (int j = 0; j < 1111; ++j) {
  98.             if (j == a[i]) {
  99.                 cnt[j][i] = cnt[j][i - 1] + 1;
  100.             } else {
  101.                 cnt[j][i] = cnt[j][i - 1];
  102.             }
  103.         }
  104.     }
  105.     //int ans = 0;
  106.     if (pf[0] == 0) {
  107.         ans++;
  108.     }
  109.     for (int i = 1; i < N; ++i) {
  110.         int S = pf[i];
  111.         //cout << "S = " << S << "\n";
  112.         int d = (1111 - S) % 1111;
  113.         ans += cnt[d][i-1];
  114.     }
  115.     cout << "ans\n";
  116.     cout << ans;
  117. }
  118.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement