結果
問題 | No.2531 Coloring Vertices on Namori |
ユーザー |
![]() |
提出日時 | 2023-11-03 22:20:48 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 113 ms / 2,000 ms |
コード長 | 3,049 bytes |
コンパイル時間 | 1,569 ms |
コンパイル使用メモリ | 134,032 KB |
実行使用メモリ | 29,888 KB |
最終ジャッジ日時 | 2024-09-25 20:44:39 |
合計ジャッジ時間 | 5,132 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 8 ms
6,816 KB |
testcase_01 | AC | 8 ms
6,940 KB |
testcase_02 | AC | 8 ms
6,940 KB |
testcase_03 | AC | 7 ms
6,940 KB |
testcase_04 | AC | 8 ms
6,940 KB |
testcase_05 | AC | 80 ms
29,780 KB |
testcase_06 | AC | 8 ms
6,944 KB |
testcase_07 | AC | 84 ms
29,796 KB |
testcase_08 | AC | 108 ms
29,668 KB |
testcase_09 | AC | 104 ms
29,888 KB |
testcase_10 | AC | 113 ms
29,808 KB |
testcase_11 | AC | 56 ms
20,348 KB |
testcase_12 | AC | 57 ms
20,476 KB |
testcase_13 | AC | 58 ms
20,352 KB |
testcase_14 | AC | 102 ms
29,788 KB |
testcase_15 | AC | 100 ms
29,808 KB |
testcase_16 | AC | 96 ms
29,844 KB |
testcase_17 | AC | 8 ms
6,940 KB |
testcase_18 | AC | 8 ms
6,940 KB |
testcase_19 | AC | 8 ms
6,940 KB |
testcase_20 | AC | 93 ms
19,288 KB |
testcase_21 | AC | 92 ms
19,204 KB |
testcase_22 | AC | 94 ms
19,320 KB |
testcase_23 | AC | 93 ms
19,288 KB |
testcase_24 | AC | 92 ms
19,196 KB |
testcase_25 | AC | 97 ms
19,168 KB |
testcase_26 | AC | 95 ms
19,212 KB |
testcase_27 | AC | 93 ms
19,220 KB |
testcase_28 | AC | 93 ms
19,224 KB |
testcase_29 | AC | 91 ms
19,108 KB |
testcase_30 | AC | 95 ms
19,208 KB |
testcase_31 | AC | 99 ms
19,168 KB |
testcase_32 | AC | 96 ms
19,208 KB |
testcase_33 | AC | 101 ms
19,264 KB |
ソースコード
#include <iostream> #include <algorithm> #include <iomanip> #include <vector> #include <queue> #include <deque> #include <set> #include <map> #include <tuple> #include <cmath> #include <numeric> #include <functional> #include <cassert> #include <atcoder/modint> #define debug_value(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << #x << "=" << x << endl; #define debug(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << x << endl; template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } using namespace std; typedef long long ll; template<typename T> vector<vector<T>> vec2d(int n, int m, T v){ return vector<vector<T>>(n, vector<T>(m, v)); } template<typename T> vector<vector<vector<T>>> vec3d(int n, int m, int k, T v){ return vector<vector<vector<T>>>(n, vector<vector<T>>(m, vector<T>(k, v))); } template<typename T> void print_vector(vector<T> v, char delimiter=' '){ if(v.empty()) { cout << endl; return; } for(int i = 0; i+1 < v.size(); i++) cout << v[i] << delimiter; cout << v.back() << endl; } using mint = atcoder::modint998244353; ostream& operator<<(ostream& os, const mint& m){ os << m.val(); return os; } #define N_MAX 200002 mint inv[N_MAX],fac[N_MAX],finv[N_MAX]; void init(){ const ll MOD = mint::mod(); fac[0]=1;fac[1]=1; finv[0]=1;finv[1]=1; inv[1]=1; for(int i=2;i<N_MAX;i++){ inv[i]=mint(MOD)-inv[MOD%i]*(MOD/i); fac[i]=fac[i-1]*(ll) i; finv[i]=finv[i-1]*inv[i]; } } mint comb(ll n, ll r){ if(n < r){ return mint(0); }else{ return fac[n]*finv[r]*finv[n-r]; } } int main(){ ios::sync_with_stdio(false); cin.tie(0); cout << setprecision(10) << fixed; init(); int n, k; cin >> n >> k; vector<int> u(n), v(n); vector<vector<int>> g(n); for(int i = 0; i < n; i++){ cin >> u[i] >> v[i]; u[i]--; v[i]--; g[u[i]].push_back(v[i]); g[v[i]].push_back(u[i]); } vector<int> deg(n); for(int i = 0; i < n; i++) deg[i] = g[i].size(); queue<int> que; vector<bool> del(n); for(int i = 0; i < n; i++){ if(deg[i] == 1){ del[i] = true; que.push(i); } } while(!que.empty()){ int v = que.front(); que.pop(); for(int to: g[v]){ if(!del[to]){ deg[to]--; if(deg[to] == 1){ del[to] = true; que.push(to); } } } } int n_cycle = 0; for(int i = 0; i < n; i++){ if(!del[i]) n_cycle++; } auto dp = vec2d(n_cycle, 2, mint(0)); dp[0][0] = k; dp[0][1] = 0; for(int i = 0; i+1 < n_cycle; i++){ dp[i+1][0] += dp[i][1]; dp[i+1][1] += dp[i][0]*(k-1) + dp[i][1]*(k-2); } mint ans = dp[n_cycle-1][1]*mint(k-1).pow(n-n_cycle); cout << ans << endl; }