結果
問題 | No.2531 Coloring Vertices on Namori |
ユーザー |
|
提出日時 | 2023-11-03 22:02:11 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 111 ms / 2,000 ms |
コード長 | 1,468 bytes |
コンパイル時間 | 663 ms |
コンパイル使用メモリ | 73,852 KB |
実行使用メモリ | 18,792 KB |
最終ジャッジ日時 | 2024-09-25 20:21:08 |
合計ジャッジ時間 | 4,469 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
#include<iostream>#include<vector>#include<cassert>#include<atcoder/modint>using namespace std;#include<vector>template<typename T>struct combination{vector<T>fac,ifac;combination(size_t N=0):fac(1,1),ifac(1,1){make_table(N);}void make_table(size_t N){if(fac.size()>N)return;size_t now=fac.size();N=max(N,now*2);fac.resize(N+1);ifac.resize(N+1);for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i;ifac[N]=1/fac[N];for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1);}T factorial(size_t n){make_table(n);return fac[n];}T invfac(size_t n){make_table(n);return ifac[n];}T P(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k];}T C(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k]*ifac[k];}T H(size_t n,size_t k){if(n==0)return k==0?1:0;return C(n-1+k,k);}};using mint=atcoder::modint998244353;int N,K;vector<int>G[2<<17];int deg[2<<17];int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin>>N>>K;for(int i=0;i<N;i++){int u,v;cin>>u>>v;u--,v--;deg[u]++;deg[v]++;G[u].push_back(v);G[v].push_back(u);}vector<int>Q;for(int i=0;i<N;i++)if(deg[i]==1)Q.push_back(i);for(int i=0;i<Q.size();i++){int u=Q[i];for(int v:G[u])if(--deg[v]==1)Q.push_back(v);}int loop=N-Q.size();cout<<((mint::raw(K-1).pow(loop)+(loop%2==0?K-1:1-K))*mint::raw(K-1).pow(N-loop)).val()<<endl;}