結果
問題 |
No.2148 ひとりUNO
|
ユーザー |
![]() |
提出日時 | 2023-02-09 12:26:39 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,825 bytes |
コンパイル時間 | 251 ms |
コンパイル使用メモリ | 82,516 KB |
実行使用メモリ | 82,632 KB |
最終ジャッジ日時 | 2024-07-06 16:07:22 |
合計ジャッジ時間 | 11,328 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 9 WA * 30 |
ソースコード
import sys readline = sys.stdin.readline class UnionFind(object): def __init__(self, n=1): self.par = [i for i in range(n)] self.rank = [0 for _ in range(n)] self.size = [1 for _ in range(n)] def find(self, x): if self.par[x] == x: return x else: self.par[x] = self.find(self.par[x]) return self.par[x] def union(self, x, y): x = self.find(x) y = self.find(y) if x != y: if self.rank[x] < self.rank[y]: x, y = y, x if self.rank[x] == self.rank[y]: self.rank[x] += 1 self.par[y] = x self.size[x] += self.size[y] def is_same(self, x, y): return self.find(x) == self.find(y) def get_size(self, x): x = self.find(x) return self.size[x] from bisect import * from copy import deepcopy def compress(lst): """ B: lstを座圧したリスト D: 元の値からindexを取得する辞書 """ B = [] D = dict() vals = deepcopy(lst) vals = list(set(vals)) vals.sort() for i in range(len(lst)): ind = bisect_left(vals, lst[i]) B.append(ind) for i in range(len(B)): D[lst[i]] = B[i] return B, D def f(c): if c == "R": return 0 elif c == "G": return 1 else: return 2 T = int(readline()) for _ in range(T): N = int(readline()) C, D = [0] * N, [0] * N for i in range(N): C[i], D[i] = readline().split() D, _ = compress(D) C = list(map(f, C)) U = UnionFind(3 * N + 10) for i in range(N): U.union(i, N + C[i]) U.union(i, N + 3 + D[i]) S = set() for i in range(N): S.add(U.find(i)) print("YES") if len(S) == 1 else print("NO")