結果
問題 | No.2531 Coloring Vertices on Namori |
ユーザー |
|
提出日時 | 2023-11-03 23:34:53 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 203 ms / 2,000 ms |
コード長 | 911 bytes |
コンパイル時間 | 3,874 ms |
コンパイル使用メモリ | 258,208 KB |
最終ジャッジ日時 | 2025-02-17 18:58:13 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using ll = long long; #define rep(i,n) for(int i=0;i<(int)(n);i++) using mint = atcoder::modint998244353; int main(){ int n,k; cin>>n>>k; vector<vector<int>> g(n); vector<int> tn(n,0); rep(i,n){ int u,v; cin>>u>>v; u--; v--; g.at(u).push_back(v); g.at(v).push_back(u); tn.at(u)++; tn.at(v)++; } queue<int> cn1; rep(i,n) if(tn.at(i)==1) cn1.push(i); while(!cn1.empty()){ int x=cn1.front(); cn1.pop(); for(int y:g.at(x)){ tn.at(y)--; if(tn.at(y)==1) cn1.push(y); } } mint ans=1; int lp=0; rep(i,n) if(tn.at(i)==2) lp++; ans*=mint(k-1).pow(n-lp); vector<vector<mint>> dp(lp,vector<mint>(2,0)); dp.at(0).at(0)=k; rep(i,lp-1){ dp.at(i+1).at(0)+=dp.at(i).at(1); dp.at(i+1).at(1)+=dp.at(i).at(0)*(k-1); dp.at(i+1).at(1)+=dp.at(i).at(1)*(k-2); } ans*=dp.at(lp-1).at(1); cout<<ans.val()<<endl; }