結果

問題 No.1023 Cyclic Tour
ユーザー aaaaaaaaaa2230
提出日時 2022-03-31 08:46:03
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,346 bytes
コンパイル時間 194 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 107,648 KB
最終ジャッジ日時 2024-11-16 10:31:01
合計ジャッジ時間 18,975 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 39 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import deque

class Unionfind:
     
    def __init__(self,n):
        self.uf = [-1]*n
 
    def find(self,x):
        if self.uf[x] < 0:
            return x
        else:
            self.uf[x] = self.find(self.uf[x])
            return self.uf[x]
 
    def same(self,x,y):
        return self.find(x) == self.find(y)
 
    def union(self,x,y):
        x = self.find(x)
        y = self.find(y)
        if x == y:
            return False
        if self.uf[x] > self.uf[y]:
            x,y = y,x
        self.uf[x] += self.uf[y]
        self.uf[y] = x
        return True
 
    def size(self,x):
        x = self.find(x)
        return -self.uf[x]

n,m = map(int,input().split())
R = [[int(x)-1 for x in input().split()] for i in range(m)]
uf = Unionfind(n)
for a,b,c in R:
    if c == 0:
        if uf.same(a,b):
            print("Yes")
            exit()
        uf.union(a,b)

e = [[] for i in range(n)]
for a,b,c in R:
    if c:
        e[uf.find(a)].append(uf.find(b))
use = [-1]*n
for i in range(n):
    if use[i] != -1:
        continue
    use[i] = i
    q = deque([i])
    while q:
        now = q.pop()
        for nex in e[now]:
            if use[nex] == i:
                print("Yes")
                exit()
            if use[nex] == -1:
                use[nex] = i
                q.append(nex)
print("No")
0