Submission #483320

#TimeUsernameProblemLanguageResultExecution timeMemory
483320BERNARB01Lottery (CEOI18_lot)C++17
65 / 100
1610 ms65540 KiB
#include <bits/stdc++.h>

using namespace std;

const int N = 10004;
const int b = (int) 1e9 + 7;
const int mod = (int) 1e9 + 9;
const int b2 = (int) 1e9 + 9;
const int mod2 = (int) 1e9 + 21;

int n, l, a[N], q;
long long pb[N], h[N];
long long pb2[N], h2[N];
vector<int> v[N];

inline pair<int, int> get(int ql, int qr) {
  return {
    pb[n - ql] * 1LL * (h[qr + 1] - h[ql] + mod) % mod,
    pb2[n - ql] * 1LL * (h2[qr + 1] - h2[ql] + mod2) % mod2
  };
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cin >> n >> l;
  for (int i = 0; i < n; i++) {
    cin >> a[i];
  }
  cin >> q;
  if (q == 1) {
    int k;
    cin >> k;
    if (k == 0) {
      map<pair<int, int>, int> mp;
      pb[0] = 1;
      pb2[0] = 1;
      for (int i = 1; i <= n; i++) {
        pb[i] = pb[i - 1] * 1LL * b % mod;
        pb2[i] = pb2[i - 1] * 1LL * b2 % mod2;
      }
      for (int i = 0; i < n; i++) {
        h[i + 1] = (pb[i] * 1LL * a[i] % mod + h[i]) % mod;
        h2[i + 1] = (pb2[i] * 1LL * a[i] % mod2 + h2[i]) % mod2;
      }
      for (int i = 0; i + l <= n; i++) {
        mp[get(i, i + l - 1)] += 1;
      }
      for (int i = 0; i + l <= n; i++) {
        if (i > 0) {
          cout << " ";
        }
        cout << mp[get(i, i + l - 1)] - 1;
      }
      cout << '\n';
    } else {
      for (int i = 0; i + l <= n; i++) {
        for (int j = i + 1; j + l <= n; j++) {
          if (j == i) continue;
          int c = 0;
          for (int kk = 0; kk < l; kk++) {
            c += (a[i + kk] != a[j + kk]);
          }
          v[i].push_back(c);
          v[j].push_back(c);
        }
      }
      for (int i = 0; i + l <= n; i++) {
        sort(v[i].begin(), v[i].end());
      }
      for (int i = 0; i + l <= n; i++) {
        if (i > 0) {
          cout << " ";
        }
        cout << upper_bound(v[i].begin(), v[i].end(), k) - v[i].begin();
      }
      cout << '\n';
    }
    return 0;
  }
  for (int i = 0; i + l <= n; i++) {
    for (int j = i + 1; j + l <= n; j++) {
      if (j == i) continue;
      int c = 0;
      for (int k = 0; k < l; k++) {
        c += (a[i + k] != a[j + k]);
      }
      v[i].push_back(c);
      v[j].push_back(c);
    }
  }
  for (int i = 0; i + l <= n; i++) {
    sort(v[i].begin(), v[i].end());
  }
  while (q--) {
    int k;
    cin >> k;
    for (int i = 0; i + l <= n; i++) {
      if (i > 0) {
        cout << " ";
      }
      cout << upper_bound(v[i].begin(), v[i].end(), k) - v[i].begin();
    }
    cout << '\n';
  }
  return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...