結果
問題 |
No.1023 Cyclic Tour
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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")