結果

問題 No.2531 Coloring Vertices on Namori
コンテスト
ユーザー rlangevin
提出日時 2023-11-03 22:50:46
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 853 bytes
コンパイル時間 165 ms
コンパイル使用メモリ 82,944 KB
実行使用メモリ 107,164 KB
最終ジャッジ日時 2024-09-25 21:13:54
合計ジャッジ時間 9,408 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N, K = map(int, input().split())
G = [[] for i in range(N)]
for i in range(N):
    u, v = map(int, input().split())
    u, v = u - 1, v - 1
    G[u].append(v)
    G[v].append(u)
    
from collections import *
Q = deque([0])
dist = [-1] * N
par = [-1] * N
loop = -1
dist[0] = 0
while Q and loop == -1:
    u = Q.popleft()
    for v in G[u]:
        if v == par[u]:
            continue
        if dist[v] != -1:
            loop = dist[u] + dist[v] + 1
            break
        par[v] = u
        dist[v] = dist[u] + 1
        Q.append(v)
        
pre0, pre1 = K, 0
mod = 998244353
for _ in range(loop - 2):
    dp0 = pre1
    dp1 = pre0 * (K - 1) + pre1 * (K - 2)
    pre0 = dp0 % mod
    pre1 = dp1 % mod    

ans = pre0 * pow(K - 1, N - loop + 1, mod) + pre1 * (K - 2) * pow(K - 1, N - loop, mod)
print(ans%mod)
0