QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#783222 | #9782. NonZero PrefSuf Sums | ucup-team3215 | AC ✓ | 11ms | 5688kb | C++20 | 2.0kb | 2024-11-26 01:54:53 | 2024-11-26 01:54:54 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
constexpr int N = 100, S = N * N;
int mod;
auto& add(auto&& a, auto b) { return a -= (a += b) >= mod? mod: 0; }
auto& addp(auto&& a, auto b, auto c) { return a = (a + b * (uint64_t)c) % mod; }
auto& mul(auto&& a, auto... b) { return ((a = a * (uint64_t)b % mod), ...); }
int fpow(int a, int b) { int r = 1; for (; b; b /= 2, mul(a, a)) if (b % 2) mul(r, a); return r; }
int dp[N + 1][S / 2 + 1], fact[N + 1], ifact[N + 1];
int main() {
int n, m; cin >> n >> m >> mod;
for (int i = 0, f = 1; i <= N; ++i) fact[i] = f, mul(f, i + 1);
for (int i = N, f = fpow(fact[N], mod - 2); ~i; --i) ifact[i] = f, mul(f, i);
dp[0][0] = 1;
int ans = 0;
for (int x = 1; x <= m; ++x) {
for (int i = n; i--; )
for (int j = 1; j <= n - i; ++j)
for (int s = i; s <= min(i * x, n - x * j); ++s) addp(dp[i + j][s + x * j], dp[i][s], ifact[j]);
if (auto c = m / x - m / (x + 1))
for (int k = 1; k <= n; ++k)
for (int i = 1; i <= min(k, n - k); ++i) {
uint64_t cans = 0;
for (int l = max(i, k - x - 1); l < k; ++l) cans += dp[i][l];
add(ans, mul(cans % mod, ifact[k], ifact[n - i - k], c));
}
for (int k = 1; k <= n; ++k)
for (int i = 1; i <= min(n - k, k - x - 2); ++i) add(ans, mul(+dp[i][k - x - 2], ifact[k], ifact[n - i - k], mod - m / (x + 1)));
}
mul(add(mul(ans, fact[n]), mul(+m, n)), 2);
fill(dp[0] + 1, *end(dp), 0);
for (int i = 0; i < n; ++i)
for (int s = 0, t = 0; s <= min(i + 1, n / 2) * m; ++s) {
dp[i + 1][s] = t;
add(t, dp[i][s]);
if (s >= m) add(t, mod - dp[i][s - m]);
}
for (int i = 0; i < n; ++i)
for (int j = 0; j <= n - i; ++j) {
uint64_t cans = 0, s = 0;
for (s = 0; s <= min(i, j) * m; s += 18, cans %= mod)
for (int d = 0; d < 18; ++d) cans += dp[i][s + d] * (uint64_t)dp[j][s + d];
add(ans, mul(cans, ifact[i], ifact[j], ifact[n - i - j], fact[n]));
}
cout << add(mod - ans, fpow(2 * m + 1, n)) << '\n';
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 5532kb
input:
2 1 998244353
output:
2
result:
ok single line: '2'
Test #2:
score: 0
Accepted
time: 3ms
memory: 5668kb
input:
69 42 696969697
output:
378553557
result:
ok single line: '378553557'
Test #3:
score: 0
Accepted
time: 1ms
memory: 5544kb
input:
2 1 998244353
output:
2
result:
ok single line: '2'
Test #4:
score: 0
Accepted
time: 0ms
memory: 5672kb
input:
69 42 696969697
output:
378553557
result:
ok single line: '378553557'
Test #5:
score: 0
Accepted
time: 3ms
memory: 5604kb
input:
61 75 677323601
output:
34613998
result:
ok single line: '34613998'
Test #6:
score: 0
Accepted
time: 1ms
memory: 5580kb
input:
13 14 670577333
output:
41465431
result:
ok single line: '41465431'
Test #7:
score: 0
Accepted
time: 1ms
memory: 5596kb
input:
14 6 987686347
output:
37536510
result:
ok single line: '37536510'
Test #8:
score: 0
Accepted
time: 1ms
memory: 5596kb
input:
15 12 196428923
output:
29322522
result:
ok single line: '29322522'
Test #9:
score: 0
Accepted
time: 2ms
memory: 5544kb
input:
68 7 786815587
output:
149281835
result:
ok single line: '149281835'
Test #10:
score: 0
Accepted
time: 1ms
memory: 5604kb
input:
3 2 503002109
output:
82
result:
ok single line: '82'
Test #11:
score: 0
Accepted
time: 1ms
memory: 5668kb
input:
13 5 756093197
output:
415698676
result:
ok single line: '415698676'
Test #12:
score: 0
Accepted
time: 1ms
memory: 5672kb
input:
2 3 646574611
output:
30
result:
ok single line: '30'
Test #13:
score: 0
Accepted
time: 2ms
memory: 5616kb
input:
39 68 120037189
output:
43217507
result:
ok single line: '43217507'
Test #14:
score: 0
Accepted
time: 0ms
memory: 5544kb
input:
13 4 423132517
output:
360231790
result:
ok single line: '360231790'
Test #15:
score: 0
Accepted
time: 0ms
memory: 5604kb
input:
14 3 309713387
output:
94215386
result:
ok single line: '94215386'
Test #16:
score: 0
Accepted
time: 0ms
memory: 5612kb
input:
24 77 886983941
output:
211636479
result:
ok single line: '211636479'
Test #17:
score: 0
Accepted
time: 1ms
memory: 5540kb
input:
3 3 504388063
output:
270
result:
ok single line: '270'
Test #18:
score: 0
Accepted
time: 0ms
memory: 5604kb
input:
3 1 936205423
output:
8
result:
ok single line: '8'
Test #19:
score: 0
Accepted
time: 0ms
memory: 5616kb
input:
3 3 295983139
output:
270
result:
ok single line: '270'
Test #20:
score: 0
Accepted
time: 1ms
memory: 5604kb
input:
10 15 446169107
output:
149884328
result:
ok single line: '149884328'
Test #21:
score: 0
Accepted
time: 1ms
memory: 5544kb
input:
37 18 833753929
output:
592917251
result:
ok single line: '592917251'
Test #22:
score: 0
Accepted
time: 0ms
memory: 5668kb
input:
11 3 998773403
output:
860630017
result:
ok single line: '860630017'
Test #23:
score: 0
Accepted
time: 0ms
memory: 5600kb
input:
14 85 688036639
output:
347188409
result:
ok single line: '347188409'
Test #24:
score: 0
Accepted
time: 1ms
memory: 5620kb
input:
3 3 844621907
output:
270
result:
ok single line: '270'
Test #25:
score: 0
Accepted
time: 1ms
memory: 5600kb
input:
3 4 204335891
output:
620
result:
ok single line: '620'
Test #26:
score: 0
Accepted
time: 1ms
memory: 5660kb
input:
7 8 113007667
output:
58946097
result:
ok single line: '58946097'
Test #27:
score: 0
Accepted
time: 0ms
memory: 5532kb
input:
4 1 637268377
output:
22
result:
ok single line: '22'
Test #28:
score: 0
Accepted
time: 1ms
memory: 5600kb
input:
11 14 391637237
output:
303270280
result:
ok single line: '303270280'
Test #29:
score: 0
Accepted
time: 1ms
memory: 5668kb
input:
3 2 208286231
output:
82
result:
ok single line: '82'
Test #30:
score: 0
Accepted
time: 1ms
memory: 5584kb
input:
2 11 662696483
output:
462
result:
ok single line: '462'
Test #31:
score: 0
Accepted
time: 1ms
memory: 5652kb
input:
19 55 974135299
output:
887460557
result:
ok single line: '887460557'
Test #32:
score: 0
Accepted
time: 0ms
memory: 5600kb
input:
6 8 417027509
output:
23351024
result:
ok single line: '23351024'
Test #33:
score: 0
Accepted
time: 0ms
memory: 5604kb
input:
8 13 624006587
output:
353008442
result:
ok single line: '353008442'
Test #34:
score: 0
Accepted
time: 0ms
memory: 5600kb
input:
10 10 740294671
output:
79436611
result:
ok single line: '79436611'
Test #35:
score: 0
Accepted
time: 0ms
memory: 5600kb
input:
11 10 394088657
output:
161476458
result:
ok single line: '161476458'
Test #36:
score: 0
Accepted
time: 1ms
memory: 5688kb
input:
9 27 562853573
output:
135252259
result:
ok single line: '135252259'
Test #37:
score: 0
Accepted
time: 1ms
memory: 5664kb
input:
8 3 829129009
output:
5349034
result:
ok single line: '5349034'
Test #38:
score: 0
Accepted
time: 0ms
memory: 5688kb
input:
51 49 924010279
output:
815049368
result:
ok single line: '815049368'
Test #39:
score: 0
Accepted
time: 1ms
memory: 5588kb
input:
12 2 308466749
output:
223013998
result:
ok single line: '223013998'
Test #40:
score: 0
Accepted
time: 1ms
memory: 5616kb
input:
7 4 567557693
output:
4502296
result:
ok single line: '4502296'
Test #41:
score: 0
Accepted
time: 0ms
memory: 5548kb
input:
36 93 943780729
output:
13599465
result:
ok single line: '13599465'
Test #42:
score: 0
Accepted
time: 1ms
memory: 5544kb
input:
2 1 828681127
output:
2
result:
ok single line: '2'
Test #43:
score: 0
Accepted
time: 1ms
memory: 5532kb
input:
3 3 534160729
output:
270
result:
ok single line: '270'
Test #44:
score: 0
Accepted
time: 1ms
memory: 5668kb
input:
7 12 920925433
output:
453086694
result:
ok single line: '453086694'
Test #45:
score: 0
Accepted
time: 1ms
memory: 5584kb
input:
3 2 440546987
output:
82
result:
ok single line: '82'
Test #46:
score: 0
Accepted
time: 2ms
memory: 5640kb
input:
90 9 291269963
output:
72560304
result:
ok single line: '72560304'
Test #47:
score: 0
Accepted
time: 1ms
memory: 5600kb
input:
38 10 867575113
output:
165530481
result:
ok single line: '165530481'
Test #48:
score: 0
Accepted
time: 2ms
memory: 5676kb
input:
48 37 152663531
output:
135425620
result:
ok single line: '135425620'
Test #49:
score: 0
Accepted
time: 1ms
memory: 5596kb
input:
15 15 991731803
output:
102703562
result:
ok single line: '102703562'
Test #50:
score: 0
Accepted
time: 11ms
memory: 5592kb
input:
100 100 696969697
output:
313377809
result:
ok single line: '313377809'
Extra Test:
score: 0
Extra Test Passed