結果
問題 | 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 = 998244353i64n, k = read_line.split.map(&.to_i)ord = Array.new(n, 0)g = Array.new(n) { [] of Int32 }n.times dou, v = read_line.split.map { |v| v.to_i - 1 }ord[u] += 1ord[v] += 1g[u] << vg[v] << uendq = n.times.select { |i| ord[i] == 1 }.to_aqi = 0ans = 1i64while qi < q.sizecur = q[qi]g[cur].each do |a|ord[a] -= 1if ord[a] == 1q << aendendqi += 1ans *= (k - 1)ans %= MODendn -= qidp = Array.new(n) { Array.new(2, 0i64) }dp[0][0] = k1.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)) % MODendputs ans * dp[-1][1] % MOD