結果
問題 | No.1614 Majority Painting on Tree |
ユーザー | 沙耶花 |
提出日時 | 2021-07-21 23:58:30 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 2,224 ms / 5,000 ms |
コード長 | 1,812 bytes |
コンパイル時間 | 4,419 ms |
コンパイル使用メモリ | 267,368 KB |
実行使用メモリ | 20,388 KB |
最終ジャッジ日時 | 2023-09-24 20:02:48 |
合計ジャッジ時間 | 41,749 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,847 ms
20,280 KB |
testcase_01 | AC | 601 ms
20,388 KB |
testcase_02 | AC | 786 ms
15,116 KB |
testcase_03 | AC | 1,351 ms
15,104 KB |
testcase_04 | AC | 1,680 ms
13,856 KB |
testcase_05 | AC | 1,456 ms
13,792 KB |
testcase_06 | AC | 2,224 ms
13,676 KB |
testcase_07 | AC | 1,415 ms
13,780 KB |
testcase_08 | AC | 785 ms
13,784 KB |
testcase_09 | AC | 891 ms
13,644 KB |
testcase_10 | AC | 259 ms
13,724 KB |
testcase_11 | AC | 1,131 ms
13,708 KB |
testcase_12 | AC | 1,243 ms
13,768 KB |
testcase_13 | AC | 898 ms
13,652 KB |
testcase_14 | AC | 1,680 ms
13,720 KB |
testcase_15 | AC | 1,744 ms
13,724 KB |
testcase_16 | AC | 401 ms
13,588 KB |
testcase_17 | AC | 871 ms
13,712 KB |
testcase_18 | AC | 918 ms
13,256 KB |
testcase_19 | AC | 362 ms
13,224 KB |
testcase_20 | AC | 2,006 ms
13,260 KB |
testcase_21 | AC | 2,035 ms
13,156 KB |
testcase_22 | AC | 2,198 ms
13,252 KB |
testcase_23 | AC | 262 ms
13,536 KB |
testcase_24 | AC | 2,047 ms
13,140 KB |
testcase_25 | AC | 566 ms
13,220 KB |
testcase_26 | AC | 1,310 ms
13,268 KB |
testcase_27 | AC | 12 ms
7,844 KB |
testcase_28 | AC | 12 ms
7,732 KB |
testcase_29 | AC | 12 ms
7,952 KB |
testcase_30 | AC | 13 ms
7,780 KB |
testcase_31 | AC | 12 ms
7,720 KB |
testcase_32 | AC | 12 ms
7,728 KB |
testcase_33 | AC | 12 ms
7,652 KB |
testcase_34 | AC | 12 ms
7,660 KB |
testcase_35 | AC | 12 ms
7,648 KB |
testcase_36 | AC | 11 ms
7,592 KB |
testcase_37 | AC | 12 ms
7,660 KB |
testcase_38 | AC | 11 ms
7,644 KB |
testcase_39 | AC | 12 ms
7,600 KB |
testcase_40 | AC | 11 ms
7,668 KB |
testcase_41 | AC | 12 ms
7,792 KB |
testcase_42 | AC | 11 ms
7,600 KB |
testcase_43 | AC | 11 ms
7,716 KB |
testcase_44 | AC | 11 ms
7,588 KB |
testcase_45 | AC | 11 ms
7,708 KB |
testcase_46 | AC | 11 ms
7,724 KB |
testcase_47 | AC | 11 ms
7,724 KB |
testcase_48 | AC | 12 ms
7,608 KB |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf 1000000000 struct combi{ deque<mint> kaijou; deque<mint> kaijou_; combi(int n){ kaijou.push_back(1); for(int i=1;i<=n;i++){ kaijou.push_back(kaijou[i-1]*i); } mint b=kaijou[n].inv(); kaijou_.push_front(b); for(int i=1;i<=n;i++){ int k=n+1-i; kaijou_.push_front(kaijou_[0]*k); } } mint combination(int n,int r){ if(r>n)return 0; mint a = kaijou[n]*kaijou_[r]; a *= kaijou_[n-r]; return a; } mint junretsu(int a,int b){ mint x = kaijou_[a]*kaijou_[b]; x *= kaijou[a+b]; return x; } mint catalan(int n){ return combination(2*n,n)/(n+1); } }; combi Co(500000); vector<vector<int>> E; mint c; vector<mint> dp; mint INV; void dfs(int cur,int p){ rep(i,E[cur].size()){ int to = E[cur][i]; if(to==p)continue; dfs(to,cur); dp[cur] *= dp[to]; } int d = E[cur].size(); mint temp = 0; { mint X = 1; rep(i,E[cur].size())X *= c; temp += X; } { mint Y = 0; mint Temp = 1; rep(i,E[cur].size()){ int remain = d-i; if(remain > d/2){ Y += Co.combination(d,i) * Temp; } else break; Temp *= c-1; } Y *= c; temp -= Y; } temp += c; if(cur!=0)temp *= INV; dp[cur] *= temp; } int main(){ int N,C; cin>>N>>C; E.resize(N); rep(i,N-1){ int u,v; scanf("%d %d",&u,&v); u--;v--; E[u].push_back(v); E[v].push_back(u); } mint ans = 0; rep(i,C){ c = C - i; mint temp = Co.combination(C,C-i); INV = c.inv(); dp.assign(N,1); dfs(0,-1); temp *= dp[0]; //cout<<temp.val()<<endl; if(i%2==1)temp *= -1; ans += temp; } cout<<ans.val()<<endl; return 0; }