QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#22941 | #2892. 合法序列 | hy_zheng_zai_nei_juan# | AC ✓ | 15ms | 3748kb | C++20 | 1.2kb | 2022-03-11 11:32:58 | 2022-04-30 02:09:25 |
Judging History
answer
#include <bits/stdc++.h>
const int mod = 998244353;
inline int inc(int x, int y) { x += y - mod; return x + (x >> 31 & mod); }
inline int dec(int x, int y) { x -= y; return x + (x >> 31 & mod); }
inline void upd(int &x, int y) { x = inc(x, y); }
int main() {
int n, k;
std::cin >> n >> k;
const int U = 1 << (1 << k);
int full = (1 << k) - 1;
int ans = 0;
auto solve = [&](int mask) {
int m = 1 << k;
std::vector<int> a(m);
for (int i = 0; i < m; ++i)
a[i] = (mask >> i & 1);
for (int i = 0; i + k <= m; ++i) {
int u = 0;
for (int j = i; j < i + k; ++j)
u = u * 2 + a[j];
if (a[u] == 0) return 0;
}
int init_v = 0;
for (int i = m - k; i < m; ++i)
init_v = init_v * 2 + a[i];
std::vector<std::vector<int>> dp(n + 1, std::vector<int>(m));
dp[m - 1][init_v] = 1;
for (int i = m; i < n; ++i) {
for (int v = 0; v < m; ++v) {
for (int bit : {0, 1}) {
int nv = (v * 2 + bit) & full;
if (a[nv] == 1) {
upd(dp[i][nv], dp[i - 1][v]);
}
}
}
}
int ans = 0;
for (int i = 0; i < m; ++i)
upd(ans, dp[n - 1][i]);
return ans;
};
for (int mask = 0; mask < U; ++mask) {
upd(ans, solve(mask));
}
std::cout << ans << '\n';
}
詳細信息
Test #1:
score: 100
Accepted
time: 3ms
memory: 3552kb
input:
2 1
output:
1
result:
ok single line: '1'
Test #2:
score: 0
Accepted
time: 2ms
memory: 3636kb
input:
10 3
output:
27
result:
ok single line: '27'
Test #3:
score: 0
Accepted
time: 2ms
memory: 3556kb
input:
11 3
output:
46
result:
ok single line: '46'
Test #4:
score: 0
Accepted
time: 3ms
memory: 3636kb
input:
500 3
output:
402086823
result:
ok single line: '402086823'
Test #5:
score: 0
Accepted
time: 5ms
memory: 3548kb
input:
16 4
output:
268
result:
ok single line: '268'
Test #6:
score: 0
Accepted
time: 1ms
memory: 3484kb
input:
17 4
output:
520
result:
ok single line: '520'
Test #7:
score: 0
Accepted
time: 5ms
memory: 3688kb
input:
18 4
output:
939
result:
ok single line: '939'
Test #8:
score: 0
Accepted
time: 5ms
memory: 3600kb
input:
19 4
output:
1634
result:
ok single line: '1634'
Test #9:
score: 0
Accepted
time: 2ms
memory: 3556kb
input:
20 4
output:
2678
result:
ok single line: '2678'
Test #10:
score: 0
Accepted
time: 15ms
memory: 3748kb
input:
500 4
output:
905810545
result:
ok single line: '905810545'
Test #11:
score: 0
Accepted
time: 0ms
memory: 3692kb
input:
156 3
output:
158348364
result:
ok single line: '158348364'
Test #12:
score: 0
Accepted
time: 3ms
memory: 3696kb
input:
3 1
output:
2
result:
ok single line: '2'
Test #13:
score: 0
Accepted
time: 3ms
memory: 3576kb
input:
419 1
output:
309520444
result:
ok single line: '309520444'
Test #14:
score: 0
Accepted
time: 1ms
memory: 3484kb
input:
148 1
output:
981388471
result:
ok single line: '981388471'
Test #15:
score: 0
Accepted
time: 3ms
memory: 3568kb
input:
208 1
output:
17788609
result:
ok single line: '17788609'
Test #16:
score: 0
Accepted
time: 14ms
memory: 3648kb
input:
325 4
output:
534470501
result:
ok single line: '534470501'
Test #17:
score: 0
Accepted
time: 2ms
memory: 3628kb
input:
478 1
output:
275981743
result:
ok single line: '275981743'
Test #18:
score: 0
Accepted
time: 0ms
memory: 3508kb
input:
437 3
output:
448979837
result:
ok single line: '448979837'
Test #19:
score: 0
Accepted
time: 1ms
memory: 3708kb
input:
260 3
output:
249722589
result:
ok single line: '249722589'
Test #20:
score: 0
Accepted
time: 3ms
memory: 3708kb
input:
406 2
output:
762498912
result:
ok single line: '762498912'
Test #21:
score: 0
Accepted
time: 0ms
memory: 3636kb
input:
348 2
output:
718553190
result:
ok single line: '718553190'
Test #22:
score: 0
Accepted
time: 3ms
memory: 3660kb
input:
350 1
output:
496751081
result:
ok single line: '496751081'
Test #23:
score: 0
Accepted
time: 3ms
memory: 3540kb
input:
500 1
output:
154029661
result:
ok single line: '154029661'
Test #24:
score: 0
Accepted
time: 12ms
memory: 3664kb
input:
344 4
output:
864324256
result:
ok single line: '864324256'
Test #25:
score: 0
Accepted
time: 1ms
memory: 3700kb
input:
127 3
output:
306783420
result:
ok single line: '306783420'
Test #26:
score: 0
Accepted
time: 5ms
memory: 3664kb
input:
170 4
output:
138463737
result:
ok single line: '138463737'
Test #27:
score: 0
Accepted
time: 3ms
memory: 3492kb
input:
199 3
output:
988988696
result:
ok single line: '988988696'
Test #28:
score: 0
Accepted
time: 3ms
memory: 3560kb
input:
206 3
output:
992080398
result:
ok single line: '992080398'
Test #29:
score: 0
Accepted
time: 1ms
memory: 3624kb
input:
82 2
output:
755922348
result:
ok single line: '755922348'
Test #30:
score: 0
Accepted
time: 3ms
memory: 3612kb
input:
351 2
output:
285073312
result:
ok single line: '285073312'
Test #31:
score: 0
Accepted
time: 3ms
memory: 3528kb
input:
274 1
output:
609866862
result:
ok single line: '609866862'
Test #32:
score: 0
Accepted
time: 1ms
memory: 3508kb
input:
416 3
output:
49398792
result:
ok single line: '49398792'
Test #33:
score: 0
Accepted
time: 3ms
memory: 3620kb
input:
4 2
output:
2
result:
ok single line: '2'
Test #34:
score: 0
Accepted
time: 3ms
memory: 3704kb
input:
5 2
output:
4
result:
ok single line: '4'
Test #35:
score: 0
Accepted
time: 2ms
memory: 3604kb
input:
6 2
output:
7
result:
ok single line: '7'
Test #36:
score: 0
Accepted
time: 1ms
memory: 3500kb
input:
500 2
output:
255888452
result:
ok single line: '255888452'
Test #37:
score: 0
Accepted
time: 2ms
memory: 3480kb
input:
8 3
output:
8
result:
ok single line: '8'
Test #38:
score: 0
Accepted
time: 0ms
memory: 3620kb
input:
9 3
output:
15
result:
ok single line: '15'