結果

問題 No.2531 Coloring Vertices on Namori
ユーザー hirayuu_ychirayuu_yc
提出日時 2023-11-21 14:07:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 876 ms / 2,000 ms
コード長 2,111 bytes
コンパイル時間 352 ms
コンパイル使用メモリ 81,700 KB
実行使用メモリ 198,580 KB
最終ジャッジ日時 2023-11-21 14:07:38
合計ジャッジ時間 22,419 ms
ジャッジサーバーID
(参考情報)
judge11 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
55,608 KB
testcase_01 AC 42 ms
55,608 KB
testcase_02 AC 42 ms
55,608 KB
testcase_03 AC 41 ms
55,608 KB
testcase_04 AC 42 ms
55,608 KB
testcase_05 AC 501 ms
197,120 KB
testcase_06 AC 41 ms
55,608 KB
testcase_07 AC 472 ms
197,888 KB
testcase_08 AC 843 ms
198,580 KB
testcase_09 AC 876 ms
196,680 KB
testcase_10 AC 857 ms
196,540 KB
testcase_11 AC 463 ms
194,880 KB
testcase_12 AC 466 ms
195,268 KB
testcase_13 AC 465 ms
195,268 KB
testcase_14 AC 674 ms
195,808 KB
testcase_15 AC 715 ms
196,208 KB
testcase_16 AC 673 ms
195,936 KB
testcase_17 AC 42 ms
55,608 KB
testcase_18 AC 41 ms
55,608 KB
testcase_19 AC 41 ms
55,608 KB
testcase_20 AC 748 ms
170,428 KB
testcase_21 AC 722 ms
171,200 KB
testcase_22 AC 723 ms
169,772 KB
testcase_23 AC 766 ms
170,936 KB
testcase_24 AC 748 ms
171,588 KB
testcase_25 AC 726 ms
171,308 KB
testcase_26 AC 745 ms
170,812 KB
testcase_27 AC 719 ms
170,436 KB
testcase_28 AC 757 ms
169,916 KB
testcase_29 AC 725 ms
171,212 KB
testcase_30 AC 768 ms
170,436 KB
testcase_31 AC 726 ms
171,428 KB
testcase_32 AC 725 ms
171,324 KB
testcase_33 AC 761 ms
171,436 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD=998244353
class Graph():
    def __init__(self,size,directed=False):
        self.dir=directed
        self.size=size
        self.gr=[[] for i in range(size)]
        self.edges=[]
    
    def add_edge(self,u,v,status={}):
        self.gr[u].append(self.Edge(u,v,status))
        if not(self.dir):
            self.gr[v].append(self.Edge(v,u,status))
        self.edges.append(self.Edge(u,v,status))

    def node(self,v):
        return self.gr[v]
    
    class Edge():
        def __init__(self,st,to,status):
            self.st=st
            self.to=to
            self.status=status
        
        def __getitem__(self,val):
            return self.status[val]

from collections import deque
def bridge(gr):
    pre=[-1]*gr.size
    low=[-1]*gr.size
    ret=[]
    for i in range(gr.size):
        if pre[i]!=-1:
            continue
        vert=deque([[0,i,-1,-1]])
        cnt=0
        while len(vert)>0:
            step,now,bef,ed=vert[-1]
            if step==1:
                for j in gr.node(now):
                    if j.to!=bef:
                        low[now]=min(low[now],low[j.to])
                if ed!=-1 and pre[now]<=low[now]:
                    ret.append(gr.node(bef)[ed])
                vert.pop()
                continue
            if pre[now]!=-1:
                low[bef]=min(low[bef],low[now])
                vert.pop()
                continue
            pre[now]=cnt
            low[now]=cnt
            cnt+=1
            vert[-1][0]=1
            for j,k in enumerate(gr.node(now)):
                if k.to==bef:
                    continue
                if low[k.to]==-1:
                    vert.append([0,k.to,now,j])
                else:
                    low[now]=min(low[now],low[k.to])
    return ret

N,K=map(int,input().split())
namori=Graph(N)
for i in range(N):
    u,v=map(int,input().split())
    namori.add_edge(u-1,v-1)
cycle=N-len(bridge(namori))
dp=[[0,0] for i in range(cycle)]
dp[0][0]=K
for i in range(cycle-1):
    dp[i+1][0]=dp[i][1]
    dp[i+1][1]=(dp[i][0]*(K-1)+dp[i][1]*(K-2))%MOD
print((dp[-1][1]*pow(K-1,N-cycle,MOD))%MOD)
0