結果

問題 No.2531 Coloring Vertices on Namori
ユーザー rlangevinrlangevin
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 44 ms
53,504 KB
testcase_01 AC 45 ms
53,504 KB
testcase_02 AC 45 ms
53,632 KB
testcase_03 AC 45 ms
53,504 KB
testcase_04 AC 45 ms
53,888 KB
testcase_05 AC 178 ms
101,760 KB
testcase_06 AC 45 ms
53,504 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 176 ms
107,164 KB
testcase_12 AC 175 ms
107,028 KB
testcase_13 AC 178 ms
106,900 KB
testcase_14 AC 292 ms
101,376 KB
testcase_15 AC 288 ms
101,504 KB
testcase_16 AC 284 ms
101,376 KB
testcase_17 WA -
testcase_18 AC 46 ms
53,504 KB
testcase_19 AC 45 ms
53,632 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
権限があれば一括ダウンロードができます

ソースコード

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