結果
問題 | No.2531 Coloring Vertices on Namori |
ユーザー |
![]() |
提出日時 | 2023-11-03 23:32:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 301 ms / 2,000 ms |
コード長 | 757 bytes |
コンパイル時間 | 282 ms |
コンパイル使用メモリ | 82,168 KB |
実行使用メモリ | 119,028 KB |
最終ジャッジ日時 | 2024-09-25 21:33:06 |
合計ジャッジ時間 | 8,289 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
import sysinput = sys.stdin.readlineN, K = map(int, input().split())G = [[] for i in range(N)]deg = [0] * Nfor i in range(N):u, v = map(int, input().split())u, v = u - 1, v - 1G[u].append(v)G[v].append(u)deg[u] += 1deg[v] += 1Q = []for i in range(N):if deg[i] == 1:Q.append(i)loop = Nwhile Q:u = Q.pop()loop -= 1for v in G[u]:deg[v] -= 1if deg[v] == 1:Q.append(v)pre0, pre1 = K, 0mod = 998244353for _ in range(loop - 2):dp0 = pre1dp1 = pre0 * (K - 1) + pre1 * (K - 2)pre0 = dp0 % modpre1 = dp1 % modans = pre0 * pow(K - 1, N - loop + 1, mod) + pre1 * (K - 2) * pow(K - 1, N - loop, mod)print(ans%mod)