QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#721656 | #9611. 木桶效应 | wsyear | 100 ✓ | 802ms | 17984kb | C++20 | 4.8kb | 2024-11-07 16:32:34 | 2024-11-07 16:32:34 |
Judging History
answer
#include <bits/stdc++.h>
#define rep(i, j, k) for (int i = (j); i <= (k); ++i)
#define per(i, j, k) for (int i = (j); i >= (k); --i)
#define SZ(v) int((v).size())
#define ALL(v) (v).begin(),(v).end()
#define fi first
#define se second
using ll = long long;
using pii = std::pair<int, int>;
using pll = std::pair<ll, ll>;
template<class T>inline void chkmn(T &x, T y) { if (y < x) x = y; }
template<class T>inline void chkmx(T &x, T y) { if (y > x) x = y; }
using namespace std;
template <int P>
class mod_int {
using Z = mod_int;
private:
static int mo(int x) { return x < 0 ? x + P : x; }
public:
int x;
int val() const { return x; }
mod_int() : x(0) {}
template <class T>
mod_int(const T &x_) : x(x_ >= 0 && x_ < P ? static_cast<int>(x_) : mo(static_cast<int>(x_ % P))) {}
bool operator==(const Z &rhs) const { return x == rhs.x; }
bool operator!=(const Z &rhs) const { return x != rhs.x; }
Z operator-() const { return Z(x ? P - x : 0); }
Z pow(long long k) const {
Z res = 1, t = *this;
while (k) {
if (k & 1) res *= t;
if (k >>= 1) t *= t;
}
return res;
}
Z &operator++() {
x < P - 1 ? ++x : x = 0;
return *this;
}
Z &operator--() {
x ? --x : x = P - 1;
return *this;
}
Z operator++(int) {
Z ret = x;
x < P - 1 ? ++x : x = 0;
return ret;
}
Z operator--(int) {
Z ret = x;
x ? --x : x = P - 1;
return ret;
}
Z inv() const { return pow(P - 2); }
Z &operator+=(const Z &rhs) {
(x += rhs.x) >= P && (x -= P);
return *this;
}
Z &operator-=(const Z &rhs) {
(x -= rhs.x) < 0 && (x += P);
return *this;
}
Z &operator*=(const Z &rhs) {
x = 1ULL * x * rhs.x % P;
return *this;
}
Z &operator/=(const Z &rhs) { return *this *= rhs.inv(); }
#define setO(T, o) \
friend T operator o(const Z &lhs, const Z &rhs) {\
Z res = lhs; \
return res o## = rhs; \
}
setO(Z, +) setO(Z, -) setO(Z, *) setO(Z, /)
#undef setO
};
const int P = 998244353;
using Z = mod_int<P>;
const int maxn = 55;
int n, m, q, tot, ln, a[maxn][maxn], vis[maxn], id[maxn], pos[maxn], popc[1 << 10];
int cnt[maxn][maxn][1 << 10];
// 第 i 行,S 集合的特殊列内有多少个 <=j 的数
// 第 i 行,S 集合内有多少个数是被钦定的
Z fac[maxn], ivf[maxn], f[maxn][maxn][1 << 10], pw[maxn], prd[maxn][maxn], mprd[maxn][maxn];
map<int, int> mp;
Z binom(int x, int y) {
if (x < 0 || y < 0 || x < y) return 0;
return fac[x] * ivf[y] * ivf[x - y];
}
int main() {
fac[0] = 1;
rep (i, 1, maxn - 1) fac[i] = fac[i - 1] * i;
ivf[maxn - 1] = fac[maxn - 1].inv();
per (i, maxn - 1, 1) ivf[i - 1] = ivf[i] * i;
cin.tie(nullptr) -> ios::sync_with_stdio(false);
cin >> n >> m >> q;
rep (i, 1, q) {
int x, y, w;
cin >> x >> y >> w;
if (!mp.count(x)) mp[x] = ++ln;
a[mp[x]][y] = n - w + 1, vis[y] = 1;
}
rep (i, 1, n) if (vis[i]) pos[++tot] = i, id[i] = tot;
int U = (1 << tot) - 1;
rep (i, 1, n) pw[i] = Z(i).pow(m - ln);
rep (i, 1, (1 << tot) - 1) popc[i] = __builtin_popcount(i);
rep (i, 1, ln) {
rep (S, 0, (1 << tot) - 1) {
rep (j, 1, tot) if (a[i][pos[j]] && (S >> (j - 1) & 1)) cnt[i][a[i][pos[j]]][S]++;
rep (j, 1, n) cnt[i][j][S] += cnt[i][j - 1][S];
}
}
rep (l, 1, n) {
prd[l][l - 1] = 1;
rep (r, l, n) prd[l][r] = prd[l][r - 1] * r;
}
rep (l, 1, n) rep (r, 1, n) mprd[l][r] = prd[l][r].pow(m - ln);
f[0][0][0] = 1;
rep (i, 1, n) {
rep (S, 0, (1 << tot) - 1) rep (j, 0, n - tot) {
rep (k, 0, j) if (f[i - 1][j - k][S].val()) {
Z ways = binom(n - tot - j + k, k);
if (i - (j - k + popc[S] + k - 1) < 0) continue;
int L = i - (j - k + popc[S] + k - 1), R = i - (j - k + popc[S]);
if (L >= 0) ways *= mprd[L][R];
rep (o, 1, ln) {
int cur = j - k + popc[S] + cnt[o][i][U ^ S];
int L = i - cur - k + 1, R = i - cur;
if (L >= 0) ways *= prd[L][R];
if (!ways.val()) break;
}
if (ways.val()) f[i][j][S] += f[i - 1][j - k][S] * ways;
}
}
rep (d, 1, tot) {
rep (j, 0, n - tot) rep (S, 0, (1 << tot) - 1) {
if ((S >> (d - 1) & 1) || !f[i][j][S].val()) continue;
Z ways = 1;
if (i - (j + popc[S]) < 0) continue;
ways *= pw[i - (j + popc[S])];
rep (o, 1, ln) {
if (a[o][pos[d]] > i) ways = 0;
else if (!a[o][pos[d]]) ways *= (i - (j + popc[S] + cnt[o][i][U ^ S]));
if (!ways.val()) break;
}
if (ways.val()) f[i][j][S ^ (1 << (d - 1))] += f[i][j][S] * ways;
}
}
}
cout << f[n][n - tot][U].val() << '\n';
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 4
Accepted
Test #1:
score: 4
Accepted
time: 0ms
memory: 17344kb
input:
5 3 0
output:
21412920
result:
ok single line: '21412920'
Test #2:
score: 4
Accepted
time: 2ms
memory: 16020kb
input:
5 3 2 3 4 4 2 5 3
output:
847674
result:
ok single line: '847674'
Test #3:
score: 4
Accepted
time: 0ms
memory: 16976kb
input:
5 3 3 3 5 5 1 2 5 2 5 3
output:
168780
result:
ok single line: '168780'
Subtask #2:
score: 8
Accepted
Dependency #1:
100%
Accepted
Test #4:
score: 8
Accepted
time: 2ms
memory: 16536kb
input:
7 3 0
output:
160221085
result:
ok single line: '160221085'
Test #5:
score: 8
Accepted
time: 0ms
memory: 16828kb
input:
7 3 2 1 1 5 1 6 2
output:
598007855
result:
ok single line: '598007855'
Test #6:
score: 8
Accepted
time: 0ms
memory: 17316kb
input:
7 3 3 1 1 5 2 6 3 2 1 7
output:
950880222
result:
ok single line: '950880222'
Subtask #3:
score: 8
Accepted
Test #7:
score: 8
Accepted
time: 2ms
memory: 16072kb
input:
50 1 0
output:
263941435
result:
ok single line: '263941435'
Test #8:
score: 8
Accepted
time: 0ms
memory: 16640kb
input:
43 2 0
output:
136378346
result:
ok single line: '136378346'
Test #9:
score: 8
Accepted
time: 2ms
memory: 17388kb
input:
50 2 0
output:
489087596
result:
ok single line: '489087596'
Subtask #4:
score: 12
Accepted
Dependency #3:
100%
Accepted
Test #10:
score: 12
Accepted
time: 0ms
memory: 17104kb
input:
50 292247015 0
output:
226872193
result:
ok single line: '226872193'
Test #11:
score: 12
Accepted
time: 2ms
memory: 15944kb
input:
50 873009728 0
output:
63648791
result:
ok single line: '63648791'
Subtask #5:
score: 16
Accepted
Dependency #2:
100%
Accepted
Dependency #4:
100%
Accepted
Test #12:
score: 16
Accepted
time: 0ms
memory: 16912kb
input:
20 728836785 5 248289571 15 16 439110385 8 15 339467267 12 7 585491339 7 9 605518440 4 1
output:
761275883
result:
ok single line: '761275883'
Test #13:
score: 16
Accepted
time: 2ms
memory: 17060kb
input:
20 288197925 5 31379347 9 1 222153278 1 1 24531748 20 1 106427339 18 1 212338331 17 1
output:
877851586
result:
ok single line: '877851586'
Test #14:
score: 16
Accepted
time: 3ms
memory: 16356kb
input:
20 880439688 5 563540321 17 20 395236367 3 20 300712779 6 20 121406689 18 20 25890496 9 20
output:
186649553
result:
ok single line: '186649553'
Test #15:
score: 16
Accepted
time: 2ms
memory: 16244kb
input:
20 311402369 5 311293636 14 13 306211116 19 19 36858994 5 11 36858994 19 10 306211116 14 18
output:
415461922
result:
ok single line: '415461922'
Test #16:
score: 16
Accepted
time: 3ms
memory: 17708kb
input:
20 98953332 5 1075868 12 5 31161114 8 12 46790018 9 10 39214697 15 7 46790018 8 8
output:
204149614
result:
ok single line: '204149614'
Subtask #6:
score: 12
Accepted
Dependency #5:
100%
Accepted
Test #17:
score: 12
Accepted
time: 11ms
memory: 17612kb
input:
50 915702052 5 541920465 39 16 833447607 49 14 326677362 14 34 412319566 10 36 206682128 46 28
output:
783441394
result:
ok single line: '783441394'
Test #18:
score: 12
Accepted
time: 5ms
memory: 17232kb
input:
50 47879239 5 30666754 20 1 23945845 17 1 27229141 25 1 40551703 9 1 15723198 18 1
output:
546399382
result:
ok single line: '546399382'
Test #19:
score: 12
Accepted
time: 19ms
memory: 17212kb
input:
50 334191703 5 167838076 19 50 95127599 33 50 221030062 4 50 89223523 36 50 40736662 39 50
output:
242104112
result:
ok single line: '242104112'
Test #20:
score: 12
Accepted
time: 5ms
memory: 17628kb
input:
50 33538004 5 1341436 40 26 19132404 4 13 22271562 24 25 19132404 40 18 19132404 24 38
output:
509216038
result:
ok single line: '509216038'
Test #21:
score: 12
Accepted
time: 6ms
memory: 17092kb
input:
50 453197583 5 304895210 17 41 450727109 33 20 129855576 4 24 214262208 12 46 214262208 4 27
output:
929330226
result:
ok single line: '929330226'
Subtask #7:
score: 20
Accepted
Dependency #6:
100%
Accepted
Test #22:
score: 20
Accepted
time: 66ms
memory: 17656kb
input:
50 733592974 7 204969642 20 32 532101473 38 22 428067108 29 19 699479674 23 35 403248947 11 33 523610998 39 50 207894041 37 47
output:
878602112
result:
ok single line: '878602112'
Test #23:
score: 20
Accepted
time: 9ms
memory: 17444kb
input:
50 363989184 7 275675456 32 1 14440487 41 1 84675645 18 1 112753584 7 1 259289962 30 1 244964889 42 1 34667760 27 1
output:
125183718
result:
ok single line: '125183718'
Test #24:
score: 20
Accepted
time: 84ms
memory: 17932kb
input:
50 987882639 7 17913003 22 50 440453368 26 50 849090802 8 50 106136773 13 50 491841612 34 50 131376512 15 50 231961452 39 50
output:
573542486
result:
ok single line: '573542486'
Test #25:
score: 20
Accepted
time: 3ms
memory: 17356kb
input:
50 739368354 7 163377222 20 27 192827369 29 10 688310083 50 13 20398047 27 45 688310083 29 11 192827369 20 5 688310083 27 27
output:
528550901
result:
ok single line: '528550901'
Test #26:
score: 20
Accepted
time: 12ms
memory: 17236kb
input:
50 691852384 7 552436456 27 11 103126750 8 13 122888842 49 34 306609066 7 11 553011271 38 32 553011271 27 47 553011271 49 50
output:
407673475
result:
ok single line: '407673475'
Test #27:
score: 20
Accepted
time: 19ms
memory: 17204kb
input:
50 828246844 7 528561184 35 4 15522388 43 35 679004541 17 20 161133582 6 20 289045078 32 49 705209075 24 38 705209075 17 44
output:
734432988
result:
ok single line: '734432988'
Subtask #8:
score: 12
Accepted
Dependency #7:
100%
Accepted
Test #28:
score: 12
Accepted
time: 263ms
memory: 17788kb
input:
50 818455715 9 460012830 5 33 118588510 47 47 281020207 23 25 175303600 1 18 234157803 48 32 256460906 19 46 287657461 24 13 81772046 14 22 114821805 44 41
output:
863394340
result:
ok single line: '863394340'
Test #29:
score: 12
Accepted
time: 36ms
memory: 17968kb
input:
50 24969517 9 23565097 13 1 11005666 43 1 21727790 47 1 12763778 6 1 6881701 34 1 2985794 23 1 13971040 15 1 13187518 29 1 1995154 12 1
output:
755357189
result:
ok single line: '755357189'
Test #30:
score: 12
Accepted
time: 373ms
memory: 17772kb
input:
50 429772980 9 414497820 4 50 141139445 19 50 407732495 7 50 83785413 1 50 164351908 8 50 213988430 26 50 52240611 50 50 352159955 27 50 279326344 29 50
output:
247413078
result:
ok single line: '247413078'
Test #31:
score: 12
Accepted
time: 16ms
memory: 17360kb
input:
50 919810726 9 403069670 48 3 193931638 39 28 821686776 22 11 432351038 8 49 466123383 14 34 254851369 2 48 193931638 48 38 821686776 8 19 254851369 8 20
output:
481094724
result:
ok single line: '481094724'
Test #32:
score: 12
Accepted
time: 21ms
memory: 17264kb
input:
50 619619808 9 140356219 25 31 289966305 18 15 308904151 23 44 530131662 15 2 174186843 35 5 452736627 42 7 12231110 21 2 308904151 18 33 140356219 18 30
output:
194056785
result:
ok single line: '194056785'
Test #33:
score: 12
Accepted
time: 40ms
memory: 17784kb
input:
50 405638475 9 189173757 25 33 313860932 22 8 196599730 26 6 273814431 3 2 52701532 15 14 335951766 19 3 313077524 29 13 349363426 7 15 313860932 19 19
output:
435209021
result:
ok single line: '435209021'
Subtask #9:
score: 8
Accepted
Dependency #8:
100%
Accepted
Test #34:
score: 8
Accepted
time: 295ms
memory: 17976kb
input:
50 946059042 10 160844820 11 40 246397981 49 28 373026730 1 18 602144414 9 9 284028279 39 44 22155194 16 2 851925494 6 17 862159117 27 32 773018252 10 20 462257810 48 19
output:
680391855
result:
ok single line: '680391855'
Test #35:
score: 8
Accepted
time: 67ms
memory: 17732kb
input:
50 260661857 10 70899239 47 1 248286199 34 1 49459370 23 1 110741847 3 1 13147285 19 1 75609139 10 1 16001621 49 1 258636033 17 1 41404092 36 1 259336018 9 1
output:
621578303
result:
ok single line: '621578303'
Test #36:
score: 8
Accepted
time: 802ms
memory: 17916kb
input:
50 444766827 10 138606634 2 50 356874907 48 50 186818234 44 50 328471606 8 50 57922803 45 50 442175507 50 50 45412871 41 50 111164719 24 50 278665393 39 50 49548102 15 50
output:
476500313
result:
ok single line: '476500313'
Test #37:
score: 8
Accepted
time: 25ms
memory: 17412kb
input:
50 471185620 10 330350191 32 46 19091103 17 9 383913632 38 25 125638511 4 13 227803510 12 49 140314389 27 34 97824926 22 46 19091103 27 16 140314389 38 4 97824926 12 1
output:
746959227
result:
ok single line: '746959227'
Test #38:
score: 8
Accepted
time: 101ms
memory: 17748kb
input:
50 534235345 10 505481965 11 39 325061903 12 35 443938537 40 20 144968487 29 31 292692014 18 39 423914803 14 36 221863124 46 41 413488879 5 39 413488879 14 5 325061903 11 42
output:
83413094
result:
ok single line: '83413094'
Test #39:
score: 8
Accepted
time: 141ms
memory: 17984kb
input:
50 268783055 10 164040236 5 3 77416439 15 8 215664484 32 34 84807773 8 44 242418459 20 43 132116147 31 13 11369559 50 23 264864429 6 42 153908131 3 27 132116147 20 50
output:
858392519
result:
ok single line: '858392519'
Test #40:
score: 8
Accepted
time: 12ms
memory: 17188kb
input:
50 680178330 10 381317639 8 3 97592656 22 6 66456445 4 9 277935856 2 13 629083753 35 33 86588466 7 25 86588466 8 6 66456445 22 11 97592656 8 23 277935856 7 24
output:
274256402
result:
ok single line: '274256402'