QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#721656#9611. 木桶效应wsyear100 ✓802ms17984kbC++204.8kb2024-11-07 16:32:342024-11-07 16:32:34

Judging History

你现在查看的是最新测评结果

  • [2024-11-07 16:32:34]
  • 评测
  • 测评结果:100
  • 用时:802ms
  • 内存:17984kb
  • [2024-11-07 16:32:34]
  • 提交

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'