結果

問題 No.2108 Red or Blue and Purple Tree
ユーザー SumitacchanSumitacchan
提出日時 2022-08-31 20:38:09
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 836 ms / 4,000 ms
コード長 2,440 bytes
コンパイル時間 3,954 ms
コンパイル使用メモリ 244,888 KB
実行使用メモリ 51,584 KB
最終ジャッジ日時 2024-11-08 15:27:24
合計ジャッジ時間 14,313 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 681 ms
51,456 KB
testcase_01 AC 836 ms
51,456 KB
testcase_02 AC 835 ms
51,456 KB
testcase_03 AC 777 ms
51,456 KB
testcase_04 AC 785 ms
51,456 KB
testcase_05 AC 792 ms
51,456 KB
testcase_06 AC 803 ms
51,584 KB
testcase_07 AC 804 ms
51,456 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/convolution>
#include <atcoder/modint>
using namespace std;
using namespace atcoder;
struct fast_ios { fast_ios(){ cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_;
#define FOR(i, begin, end) for(int i=(begin);i<(end);i++)
#define REP(i, n) FOR(i,0,n)
#define IFOR(i, begin, end) for(int i=(end)-1;i>=(begin);i--)
#define IREP(i, n) IFOR(i,0,n)
#define debug(x) cout << #x << "=" << x << endl;
#define vdebug(v) { cout << #v << "=" << endl; REP(i_debug, (int)v.size()){ cout << v[i_debug] << ","; } cout << endl; }
#define mdebug(m) { cout << #m << "=" << endl; REP(i_debug, (int)m.size()){ REP(j_debug, (int)m[i_debug].size()){ cout << m[i_debug][j_debug] << ","; } cout << endl;} }

using mint = modint998244353;
using ll = long long;

int main(){

    const ll MAX = 2000;

    vector<mint> fact(MAX + 1), invfact(MAX + 1);
    fact[0] = 1;
    FOR(n, 1, MAX + 1) fact[n] = fact[n - 1] * n;
    invfact[MAX] = fact[MAX].inv();
    IFOR(n, 0, MAX) invfact[n] = invfact[n + 1] * (n + 1);

    //dpの遷移
    vector<mint> f(MAX + 1);
    FOR(n, 1, MAX + 1){
        f[n] = ((mint)n).pow(n) * invfact[n];
    }

    // f(x)^k (k<=MAX) を計算
    vector<vector<mint>> dp(MAX + 1);
    dp[0].resize(MAX + 1);
    dp[0][0] = 1;
    FOR(i, 1, MAX + 1){
        dp[i] = convolution(dp[i - 1], f);
        dp[i].resize(MAX + 1);
    }

    vector<vector<mint>> ans(MAX + 1, vector<mint>(MAX + 1));
    FOR(n, 2, MAX + 1){
        // n^(2*(i-1))
        vector<mint> pow_n(n);
        pow_n[0] = ((mint)(n * n)).inv();
        FOR(i, 1, n) pow_n[i] = pow_n[i - 1] * (n * n);

        vector<mint> g(n);
        REP(k, n){
            //紫はk本あり、赤青は重複しても良いとしたときの答え (非紫の辺によりn-k個の連結成分)
            g[k] = dp[n - k][n] * fact[n] * invfact[n - k] * pow_n[(n - k - 2) + 1];
            //畳み込むため補正
            g[k] *= fact[k];
            if(k % 2) g[k] = -g[k];
        }

        vector<mint> h(n);
        REP(k, n){
            h[n - 1 - k] = invfact[k];
        }

        auto gh = convolution(g, h);
        REP(k, n){
            ans[n][k] = gh[n - 1 + k] * invfact[k];
            if(k % 2) ans[n][k] = -ans[n][k];
        }
    }

    ll T; cin >> T;
    REP(i, T){
        ll N, K; cin >> N >> K;
        cout << ans[N][K].val() << '\n';
    }
    
    return 0;
}
0