結果
問題 | No.2531 Coloring Vertices on Namori |
ユーザー |
👑 ![]() |
提出日時 | 2023-11-03 22:06:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 463 ms / 2,000 ms |
コード長 | 1,428 bytes |
コンパイル時間 | 296 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 171,008 KB |
最終ジャッジ日時 | 2024-09-25 20:31:07 |
合計ジャッジ時間 | 13,700 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
"""https://yukicoder.me/problems/no/2531"""import mathimport sysfrom sys import stdindef modfac(n, MOD):f = 1factorials = [1]for m in range(1, n + 1):f *= mf %= MODfactorials.append(f)inv = pow(f, MOD - 2, MOD)invs = [1] * (n + 1)invs[n] = invfor m in range(n, 1, -1):inv *= minv %= MODinvs[m - 1] = invreturn factorials, invsdef modnCr(n,r): #上で求めたfacとinvsを引数に入れるべし(上の関数で与えたnが計算できる最大のnになる)return fac[n] * inv[n-r] * inv[r] % moddef cycle(r,K):ret = 0pp = 1for i in range(r-1):ret += pp * K * pow(K-1,r-1-i,mod)pp *= -1return ret % modmod = 998244353fac,inv = modfac(10**6+100,mod)N,K = map(int,stdin.readline().split())lis = [ [] for i in range(N) ]inlis = [0] * N#print (cycle(N,K))for i in range(N):u,v = map(int,stdin.readline().split())u -= 1v -= 1lis[u].append(v)lis[v].append(u)inlis[u] += 1inlis[v] += 1from collections import dequeq = deque()for i in range(N):if inlis[i] <= 1:q.append(i)tree = 0while q:v = q.popleft()tree += 1for nex in lis[v]:inlis[nex] -= 1if inlis[nex] == 1:q.append(nex)inlis[nex] = -1print (cycle(N-tree,K) * pow(K-1,tree,mod) % mod)