QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#555789 | #9116. DRD String | zlt | AC ✓ | 4ms | 5456kb | C++14 | 1.8kb | 2024-09-10 09:55:35 | 2024-09-10 09:55:35 |
Judging History
answer
#include <bits/stdc++.h>
#define pb emplace_back
#define fst first
#define scd second
#define mkp make_pair
#define mems(a, x) memset((a), (x), sizeof(a))
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
typedef long double ldb;
typedef pair<ll, ll> pii;
const int maxn = 40;
const ll mod = 998244353;
ll n, m;
struct mat {
ll a[maxn][maxn];
mat() {
mems(a, 0);
}
} F[maxn], G[maxn], H[maxn];
inline mat operator * (const mat &a, const mat &b) {
mat res;
for (int k = 0; k < maxn; ++k) {
for (int i = 0; i < maxn; ++i) {
for (int j = 0; j < maxn; ++j) {
res.a[i][j] = (res.a[i][j] + a.a[i][k] * b.a[k][j]) % mod;
}
}
}
return res;
}
struct vec {
ll a[maxn];
vec() {
mems(a, 0);
}
};
inline vec operator * (const vec &a, const mat &b) {
vec res;
for (int i = 0; i < maxn; ++i) {
for (int j = 0; j < maxn; ++j) {
res.a[j] = (res.a[j] + a.a[i] * b.a[i][j]) % mod;
}
}
return res;
}
void solve() {
scanf("%lld%lld", &n, &m);
for (int l = 0; l < 20; ++l) {
for (int i = 0; i < 40; ++i) {
F[l].a[i][i] = 1;
}
for (int i = 0; i <= l; ++i) {
if (i < l) {
F[l].a[i + 20][i + 20] = m;
}
for (int j = i; j < l; ++j) {
F[l].a[j][i] = F[l].a[j + 20][i] = m * (((j - i) & 1) ? mod - 1 : 1) % mod;
}
F[l].a[l][i] = m * (((l - i) & 1) ? mod - 1 : 1) % mod;
}
}
G[0] = H[0] = F[0];
for (int i = 1; i < 20; ++i) {
G[i] = H[i - 1] * F[i];
H[i] = G[i] * H[i - 1];
}
vec ans;
for (int i = 20; i < 40; ++i) {
ans.a[i] = 1;
}
--n;
for (int i = 19; ~i; --i) {
if (n & (1 << i)) {
ans = ans * G[i];
}
}
printf("%lld\n", ans.a[0] * m % mod);
}
int main() {
int T = 1;
// scanf("%d", &T);
while (T--) {
solve();
}
return 0;
}
这程序好像有点Bug,我给组数据试试?
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 4ms
memory: 5388kb
input:
6 2
output:
40
result:
ok "40"
Test #2:
score: 0
Accepted
time: 4ms
memory: 5384kb
input:
3017 7801
output:
515391664
result:
ok "515391664"
Test #3:
score: 0
Accepted
time: 0ms
memory: 5308kb
input:
3 1
output:
1
result:
ok "1"
Test #4:
score: 0
Accepted
time: 4ms
memory: 5456kb
input:
4 7
output:
343
result:
ok "343"
Test #5:
score: 0
Accepted
time: 4ms
memory: 5388kb
input:
5 4
output:
304
result:
ok "304"
Test #6:
score: 0
Accepted
time: 4ms
memory: 5376kb
input:
8 8
output:
2355200
result:
ok "2355200"
Test #7:
score: 0
Accepted
time: 4ms
memory: 5380kb
input:
7 6
output:
54216
result:
ok "54216"
Test #8:
score: 0
Accepted
time: 4ms
memory: 5264kb
input:
1330 3031
output:
139921223
result:
ok "139921223"
Test #9:
score: 0
Accepted
time: 4ms
memory: 5296kb
input:
4946 3837
output:
64102067
result:
ok "64102067"
Test #10:
score: 0
Accepted
time: 4ms
memory: 5368kb
input:
4236 3305
output:
78581604
result:
ok "78581604"
Test #11:
score: 0
Accepted
time: 0ms
memory: 5372kb
input:
399 3245
output:
714500544
result:
ok "714500544"
Test #12:
score: 0
Accepted
time: 4ms
memory: 5320kb
input:
4881 2346
output:
28365995
result:
ok "28365995"
Test #13:
score: 0
Accepted
time: 4ms
memory: 5328kb
input:
4647 3069
output:
798067847
result:
ok "798067847"
Test #14:
score: 0
Accepted
time: 2ms
memory: 5260kb
input:
3414 4280
output:
669878613
result:
ok "669878613"
Test #15:
score: 0
Accepted
time: 4ms
memory: 5456kb
input:
1000000 1000000
output:
469217978
result:
ok "469217978"
Test #16:
score: 0
Accepted
time: 4ms
memory: 5320kb
input:
1000000 1
output:
1
result:
ok "1"
Test #17:
score: 0
Accepted
time: 4ms
memory: 5304kb
input:
418909 302156
output:
189551565
result:
ok "189551565"
Test #18:
score: 0
Accepted
time: 0ms
memory: 5256kb
input:
984572 895728
output:
567020798
result:
ok "567020798"
Test #19:
score: 0
Accepted
time: 4ms
memory: 5264kb
input:
50545 723857
output:
651400574
result:
ok "651400574"
Test #20:
score: 0
Accepted
time: 4ms
memory: 5240kb
input:
668791 286918
output:
27505324
result:
ok "27505324"
Test #21:
score: 0
Accepted
time: 4ms
memory: 5384kb
input:
898539 238506
output:
9072376
result:
ok "9072376"
Test #22:
score: 0
Accepted
time: 4ms
memory: 5308kb
input:
711024 686568
output:
221989803
result:
ok "221989803"
Test #23:
score: 0
Accepted
time: 4ms
memory: 5452kb
input:
970385 660016
output:
104842944
result:
ok "104842944"
Test #24:
score: 0
Accepted
time: 0ms
memory: 5300kb
input:
943632 224593
output:
631779281
result:
ok "631779281"
Test #25:
score: 0
Accepted
time: 4ms
memory: 5300kb
input:
26758 836991
output:
860673946
result:
ok "860673946"
Test #26:
score: 0
Accepted
time: 4ms
memory: 5376kb
input:
838661 823355
output:
34634389
result:
ok "34634389"
Test #27:
score: 0
Accepted
time: 4ms
memory: 5260kb
input:
923587 8580
output:
549615557
result:
ok "549615557"
Test #28:
score: 0
Accepted
time: 0ms
memory: 5300kb
input:
275456 449218
output:
745638801
result:
ok "745638801"
Test #29:
score: 0
Accepted
time: 4ms
memory: 5376kb
input:
96512 594328
output:
159367796
result:
ok "159367796"
Test #30:
score: 0
Accepted
time: 4ms
memory: 5260kb
input:
995445 996221
output:
785170205
result:
ok "785170205"
Test #31:
score: 0
Accepted
time: 4ms
memory: 5376kb
input:
821123 195922
output:
950462887
result:
ok "950462887"
Test #32:
score: 0
Accepted
time: 4ms
memory: 5268kb
input:
715365 748511
output:
571984302
result:
ok "571984302"
Test #33:
score: 0
Accepted
time: 4ms
memory: 5388kb
input:
867858 338674
output:
698272834
result:
ok "698272834"
Test #34:
score: 0
Accepted
time: 0ms
memory: 5380kb
input:
406213 206504
output:
201893603
result:
ok "201893603"
Test #35:
score: 0
Accepted
time: 4ms
memory: 5388kb
input:
729455 262034
output:
521537323
result:
ok "521537323"
Test #36:
score: 0
Accepted
time: 4ms
memory: 5368kb
input:
915061 381221
output:
361349035
result:
ok "361349035"
Test #37:
score: 0
Accepted
time: 4ms
memory: 5320kb
input:
426669 32501
output:
206011209
result:
ok "206011209"
Test #38:
score: 0
Accepted
time: 4ms
memory: 5300kb
input:
256786 708325
output:
287335753
result:
ok "287335753"
Extra Test:
score: 0
Extra Test Passed