結果

問題 No.3004 ヤング図形
ユーザー Tatsu_mr
提出日時 2025-01-18 19:27:42
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,014 bytes
コンパイル時間 3,548 ms
コンパイル使用メモリ 293,968 KB
実行使用メモリ 814,612 KB
最終ジャッジ日時 2025-01-18 19:28:23
合計ジャッジ時間 32,939 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 MLE * 1
other AC * 21 TLE * 1 MLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;

#define For(i, a, b) for(int i = (a); i < (b); i++)
#define rep(i, n) For(i, 0, n)
#define rFor(i, a, b) for(int i = (a); i >= (b); i--)
#define ALL(v) (v).begin(), (v).end()
#define rALL(v) (v).rbegin(), (v).rend()

using lint = long long;
using ld = long double;

int INF = 2000000000;
lint LINF = 1000000000000000000;

using mint = atcoder::modint998244353;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int k;
    cin >> k;
    vector<int> l(k), m(k);
    int n = 0;
    rep(i, k) {
        cin >> l[i] >> m[i];
        n += l[i] * m[i];
    }
    
    vector<int> num;
    int sum = n;
    rep(i, k) {
        num.emplace_back(sum);
        num.emplace_back(l[i] * m[i]);
        num.emplace_back(sum - l[i] * m[i]);
        sum -= l[i] * m[i];
    }
    rep(i, k) {
        int rest = l[i] * m[i];
        rep(_, m[i]) {
            num.emplace_back(rest);
            num.emplace_back(l[i]);
            num.emplace_back(rest - l[i]);
            rest -= l[i];
        }
        num.emplace_back(m[i]);
    }
    sort(ALL(num));
    num.erase(unique(ALL(num)), num.end());
    map<int, mint> Fac, Finv;
    Fac[0] = Fac[1] = 1;
    Finv[0] = Finv[1] = 1;
    mint now = 1;
    int mx = num.back();
    int cur = 0;
    while (num[cur] < 2) {
        cur++;
    }
    for (int i = 2; i <= mx; i++) {
        now *= mint(i);
        if (cur < (int)num.size() && num[cur] == i) {
            Fac[i] = now;
            Finv[i] = now.inv();
            cur++;
        }
    }
    
    auto comb = [&](int n, int k) -> mint {
        return Fac[n] * Finv[k] * Finv[n - k];
    };
    
    mint ans = 1;
    sum = n;
    rep(i, k) {
        ans *= comb(sum, l[i] * m[i]);
        sum -= l[i] * m[i];
    }
    rep(i, k) {
        int rest = l[i] * m[i];
        rep(_, m[i]) {
            ans *= comb(rest, l[i]);
            rest -= l[i];
        }
        ans /= Fac[m[i]];
    }
    cout << ans.val() << "\n";
}
0