結果
問題 | No.2531 Coloring Vertices on Namori |
ユーザー |
![]() |
提出日時 | 2023-11-03 22:11:43 |
言語 | Crystal (1.14.0) |
結果 |
AC
|
実行時間 | 229 ms / 2,000 ms |
コード長 | 655 bytes |
コンパイル時間 | 12,883 ms |
コンパイル使用メモリ | 297,920 KB |
実行使用メモリ | 37,136 KB |
最終ジャッジ日時 | 2024-09-25 20:37:22 |
合計ジャッジ時間 | 19,008 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
MOD = 998244353i64 n, k = read_line.split.map(&.to_i) ord = Array.new(n, 0) g = Array.new(n) { [] of Int32 } n.times do u, v = read_line.split.map { |v| v.to_i - 1 } ord[u] += 1 ord[v] += 1 g[u] << v g[v] << u end q = n.times.select { |i| ord[i] == 1 }.to_a qi = 0 ans = 1i64 while qi < q.size cur = q[qi] g[cur].each do |a| ord[a] -= 1 if ord[a] == 1 q << a end end qi += 1 ans *= (k - 1) ans %= MOD end n -= qi dp = Array.new(n) { Array.new(2, 0i64) } dp[0][0] = k 1.upto(n - 1) do |i| dp[i][0] = dp[i - 1][1] dp[i][1] = (dp[i - 1][0] * (k - 1) + dp[i - 1][1] * (k - 2)) % MOD end puts ans * dp[-1][1] % MOD