結果

問題 No.3426 Mod K Graph Increments (Hard)
コンテスト
ユーザー prd_xxx
提出日時 2026-01-11 16:11:38
言語 PyPy3
(7.3.17)
結果
WA  
実行時間 -
コード長 2,135 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 232 ms
コンパイル使用メモリ 82,616 KB
実行使用メモリ 149,076 KB
最終ジャッジ日時 2026-01-11 16:11:41
合計ジャッジ時間 3,150 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 4 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

class UnionFind:
    def __init__(self,N):
        self.parent = [i for i in range(N)]
        self._size = [1] * N
        self.count = 0
    def root(self,a):
        if self.parent[a] == a:
            return a
        else:
            self.parent[a] = self.root(self.parent[a])
            return self.parent[a]
    def is_same(self,a,b):
        return self.root(a) == self.root(b)
    def unite(self,a,b):
        ra = self.root(a)
        rb = self.root(b)
        if ra == rb: return
        if self._size[ra] < self._size[rb]: ra,rb = rb,ra
        self._size[ra] += self._size[rb]
        self.parent[rb] = ra
        self.count += 1
    def size(self,a):
        return self._size[self.root(a)]


import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**8)

def solve(N,M,K,UV,B):
    es = [[] for _ in range(N)]
    for u,v in UV:
        u,v = u-1,v-1
        es[u].append(v)
        es[v].append(u)
    
    if M==N-1:
        A = [0] * N
        def rec(v,p=-1):
            for to in es[v]:
                if to==p: continue
                A[v] += rec(to,v)
                A[v] %= K
            ret = (B[v] - A[v]) % K
            return ret
        rec(0)
        return 'Yes' if A[0]==B[0] else 'No'
    
    ST = [[] for _ in range(N)]
    rem = []
    uf = UnionFind(N)
    for u,v in UV:
        u,v = u-1,v-1
        if uf.is_same(u,v):
            rem.append((u,v))
        else:
            uf.unite(u,v)
            ST[u].append(v)
            ST[v].append(u)
    
    costs = [0] * N
    costs[0] = B[0]
    visited = [0] * N
    visited[0] = 1
    stack = [0]
    while stack:
        v = stack.pop()
        for to in ST[v]:
            if visited[to]: continue
            visited[to] = 1
            costs[to] = costs[v] + B[to]
            stack.append(to)
    if K%2==0:
        for u,v in rem:
            if (costs[u] + costs[v]) % 2:
                return 'No'
    return 'Yes'
    


T = int(input())
for _ in range(T):
    N,M,K = map(int,input().split())
    UV = [tuple(map(int,input().split())) for _ in range(M)]
    B = list(map(int,input().split()))
    print(solve(N,M,K,UV,B))
0