#include <bits/stdc++.h>
using namespace std;
const int N = 10000, Q = 101;
int v[2 * N], map_q[N], ant[Q][N];
vector<int> qs;
int n, l, q;
int main() {
#ifdef HOME
freopen("lot.in", "r", stdin);
freopen("lot.out", "w", stdout);
#endif
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
cin >> n >> l;
for (int i = 0; i < n; ++i) {
cin >> v[i];
v[i + n] = v[i]; }
cin >> q;
qs.resize(q);
for (int i = 0; i < q; ++i) {
cin >> qs[i];
map_q[qs[i]] = i + 1; }
for (int i = N - 2; i >= 0; --i)
if (!map_q[i])
map_q[i] = map_q[i + 1];
for (int shift = 1; shift < n; ++shift) {
int dif = 0, st = shift, dr = shift;
for (int i = 0; i < l; ++i) {
dif+= int(v[dr] != v[i]);
dr = (dr + 1 == n ? 0 : dr + 1); }
if (st <= (dr == 0 ? n - 1 : dr - 1))
ant[map_q[dif]][0]+= 1;
for (int i = 1; i < n - l + 1; ++i) {
dif-= int(v[st] != v[i - 1]);
st = (st + 1 == n ? 0 : st + 1);
dif+= int(v[dr] != v[i + l - 1]);
dr = (dr + 1 == n ? 0 : dr + 1);
if (st <= (dr == 0 ? n - 1 : dr - 1))
ant[map_q[dif]][i]+= 1; } }
for (int i = 1; i <= q; ++i)
for (int j = 0; j < n - l + 1; ++j)
ant[i][j]+= ant[i - 1][j];
for (int i = 0; i < q; ++i) {
for (int j = 0; j < n - l + 1; ++j)
cout << ant[map_q[qs[i]]][j] << ' ' ;
cout << '\n'; }
return 0; }
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Incorrect |
3 ms |
512 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Incorrect |
3 ms |
512 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
380 ms |
636 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
380 ms |
636 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Incorrect |
3 ms |
512 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |