제출 #530339

#제출 시각아이디문제언어결과실행 시간메모리
530339abc864197532Lottery (CEOI18_lot)C++17
100 / 100
1844 ms8356 KiB
#include <bits/stdc++.h> using namespace std; #define lli long long int #define mp make_pair #define pb push_back #define eb emplace_back #define X first #define Y second #define pii pair<int,int> #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() void abc() {cout << endl;} template <typename T, typename ...U> void abc(T i, U ...j) { cout << i << ' ', abc(j...); } template <typename T> void printv(T l, T r) { for (; l != r; ++l) cout << *l << " \n"[l + 1 == r]; } #ifdef Doludu #define test(x...) abc("[" + string(#x) + "]", x); #define owo freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout) #else #define test(x...) void(0); #define owo ios::sync_with_stdio(false), cin.tie(0) #endif const int N = 200000; int main () { owo; int n, len; cin >> n >> len; vector <int> v(n); for (int i = 0; i < n; ++i) cin >> v[i]; int q; cin >> q; vector <int> queries(q); vector <vector <int>> ans(q, vector <int>(n - len + 1)); for (int i = 0; i < q; ++i) cin >> queries[i]; vector <int> id(q); iota(all(id), 0), sort(all(id), [&](int i, int j) { return queries[i] < queries[j]; }); vector <int> res; for (int d = 1; len + d <= n; ++d) { res.assign(n + 1, 0); int bound = max(n - d - len + 1, 0); for (int a = 0; a + d <= n; ++a) { int b = a + d; if (v[a] != v[b]) continue; // x + i = a, y + i = b, a - len < st <= a // y - x = b - a // a - len + 1 <= st < a + 1 res[max(a - len + 1, 0)]++; res[min(a + 1, bound)]--; } vector <int> cnt(n + 1, 0), ord(n - d - len + 1); for (int i = 1; i + d + len <= n; ++i) res[i] += res[i - 1]; for (int i = 0; i + d + len <= n; ++i) cnt[res[i]]++; for (int i = 1; i <= n; ++i) cnt[i] += cnt[i - 1]; for (int i = 0; i + d + len <= n; ++i) { ord[--cnt[res[i]]] = i; } for (int i = n - d - len, j = 0; j < q; ++j) { while (~i && res[ord[i]] >= len - queries[id[j]]) ans[id[j]][ord[i]]++, ans[id[j]][ord[i] + d]++, i--; } } for (int i = 1; i < q; ++i) { for (int j = 0; j + len <= n; ++j) ans[id[i]][j] += ans[id[i - 1]][j]; } for (int i = 0; i < q; ++i) printv(all(ans[i])); }
#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...