結果

問題 No.1023 Cyclic Tour
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2022-03-31 08:52:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 441 ms / 2,000 ms
コード長 1,283 bytes
コンパイル時間 366 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 109,056 KB
最終ジャッジ日時 2024-11-16 10:39:02
合計ジャッジ時間 21,171 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

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)]
out = [0]*n
for a,b,c in R:
    if c:
        e[uf.find(b)].append(uf.find(a))
        out[uf.find(a)] += 1
q = deque([])
for i in range(n):
    if out[i]:
        continue
    q.append(i)

while q:
    now = q.popleft()
    for nex in e[now]:
        out[nex] -= 1
        if out[nex] == 0:
            q.append(nex)
print("Yes" if sum(out) else "No")

0