結果

問題 No.1614 Majority Painting on Tree
ユーザー momoyuumomoyuu
提出日時 2024-08-09 21:02:52
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,487 ms / 5,000 ms
コード長 1,673 bytes
コンパイル時間 1,444 ms
コンパイル使用メモリ 115,032 KB
実行使用メモリ 14,592 KB
最終ジャッジ日時 2024-08-09 21:03:20
合計ジャッジ時間 22,715 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,197 ms
14,592 KB
testcase_01 AC 373 ms
14,592 KB
testcase_02 AC 399 ms
11,648 KB
testcase_03 AC 605 ms
11,648 KB
testcase_04 AC 653 ms
10,916 KB
testcase_05 AC 627 ms
10,752 KB
testcase_06 AC 923 ms
10,880 KB
testcase_07 AC 654 ms
10,752 KB
testcase_08 AC 324 ms
10,880 KB
testcase_09 AC 427 ms
10,624 KB
testcase_10 AC 132 ms
10,620 KB
testcase_11 AC 542 ms
10,624 KB
testcase_12 AC 560 ms
10,752 KB
testcase_13 AC 469 ms
10,624 KB
testcase_14 AC 842 ms
10,696 KB
testcase_15 AC 781 ms
10,624 KB
testcase_16 AC 200 ms
10,496 KB
testcase_17 AC 411 ms
10,624 KB
testcase_18 AC 552 ms
10,368 KB
testcase_19 AC 236 ms
10,368 KB
testcase_20 AC 1,300 ms
10,328 KB
testcase_21 AC 1,354 ms
10,240 KB
testcase_22 AC 1,487 ms
10,368 KB
testcase_23 AC 170 ms
10,240 KB
testcase_24 AC 1,323 ms
10,368 KB
testcase_25 AC 313 ms
10,368 KB
testcase_26 AC 832 ms
10,240 KB
testcase_27 AC 3 ms
6,940 KB
testcase_28 AC 3 ms
6,940 KB
testcase_29 AC 2 ms
6,944 KB
testcase_30 AC 2 ms
6,944 KB
testcase_31 AC 2 ms
6,940 KB
testcase_32 AC 3 ms
6,944 KB
testcase_33 AC 2 ms
6,944 KB
testcase_34 AC 2 ms
6,940 KB
testcase_35 AC 2 ms
6,944 KB
testcase_36 AC 2 ms
6,944 KB
testcase_37 AC 2 ms
6,940 KB
testcase_38 AC 2 ms
6,944 KB
testcase_39 AC 2 ms
6,944 KB
testcase_40 AC 2 ms
6,944 KB
testcase_41 AC 2 ms
6,940 KB
testcase_42 AC 2 ms
6,940 KB
testcase_43 AC 1 ms
6,940 KB
testcase_44 AC 1 ms
6,940 KB
testcase_45 AC 2 ms
6,944 KB
testcase_46 AC 2 ms
6,944 KB
testcase_47 AC 1 ms
6,944 KB
testcase_48 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;
using ll = long long;

#include<atcoder/modint>
using mint = atcoder::modint998244353;

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    int n,c;
    cin>>n>>c;
    vector<vector<int>> g(n);
    for(int i = 0;i<n-1;i++){
        int u,v;
        cin>>u>>v;
        u--;v--;
        g[u].push_back(v);
        g[v].push_back(u);
    }

    int nni = 1;
    vector<mint> nipow(n+1,1),njpow(n+1,1);
    mint inv;
    vector<mint> fac(n+1,0),ifac(n+1,0);
    fac[0] = 1;
    for(int i = 1;i<=n;i++) fac[i] = fac[i-1] * i;
    ifac[n] = fac[n].inv();
    for(int i =  n-1;i>=0;i--) ifac[i] = ifac[i+1] * (i+1);
    auto dfs = [&](auto dfs,int ni,int p) -> mint {
        int all = g[ni].size();
        mint sum = 1;
        for(auto&to:g[ni]) if(to!=p) {
            mint tmp = dfs(dfs,to,ni);
            sum *= tmp;
        }
        mint ans = nipow[all] * sum;
        for(int i = 1;i<all;i++){
            if(2*i<=all) continue;
            mint tmp = fac[all] * ifac[i] * ifac[all-i];
            tmp *= nni;
            tmp *= njpow[all-i];
            tmp *= sum;
            ans -= tmp;
        }
        return ans * inv;
    };

    mint ans = 0;
    for(int i = 1;i<=c;i++){
        nni = i;
        inv = mint(nni).inv();
        for(int j = 1;j<=n;j++){
            nipow[j] = nipow[j-1] * nni;
            njpow[j] = njpow[j-1] * (nni-1);
        }
        auto use = dfs(dfs,0,-1);
        use *= nni;
        if((c-i)%2==0) ans += use * fac[c] * ifac[i] * ifac[c-i];
        else ans -= use * fac[c] * ifac[i] * ifac[c-i];
    }
    cout<<ans.val()<<endl;
}
0