結果
| 問題 | No.2290 UnUnion Find |
| ユーザー |
 NP
|
| 提出日時 | 2024-07-04 21:36:46 |
| 言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,852 bytes |
| コンパイル時間 | 196 ms |
| コンパイル使用メモリ | 12,928 KB |
| 実行使用メモリ | 48,012 KB |
| 最終ジャッジ日時 | 2024-07-04 21:36:53 |
| 合計ジャッジ時間 | 5,454 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge5 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 32 ms
17,820 KB |
| testcase_01 | AC | 29 ms
10,880 KB |
| testcase_02 | TLE | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
| testcase_26 | -- | - |
| testcase_27 | -- | - |
| testcase_28 | -- | - |
| testcase_29 | -- | - |
| testcase_30 | -- | - |
| testcase_31 | -- | - |
| testcase_32 | -- | - |
| testcase_33 | -- | - |
| testcase_34 | -- | - |
| testcase_35 | -- | - |
| testcase_36 | -- | - |
| testcase_37 | -- | - |
| testcase_38 | -- | - |
| testcase_39 | -- | - |
| testcase_40 | -- | - |
| testcase_41 | -- | - |
| testcase_42 | -- | - |
| testcase_43 | -- | - |
| testcase_44 | -- | - |
| testcase_45 | -- | - |
| testcase_46 | -- | - |
ソースコード
class DSU:
def __init__(self, n):
self.parent = list(range(n))
self.rank = [1] * n
def find(self, x):
if self.parent[x] != x:
self.parent[x] = self.find(self.parent[x])
return self.parent[x]
def union(self, x, y):
rootX = self.find(x)
rootY = self.find(y)
if rootX != rootY:
if self.rank[rootX] > self.rank[rootY]:
self.parent[rootY] = rootX
elif self.rank[rootX] < self.rank[rootY]:
self.parent[rootX] = rootY
else:
self.parent[rootY] = rootX
self.rank[rootX] += 1
def process_queries(N, Q, queries):
dsu = DSU(N)
connected = [set() for _ in range(N)]
results = []
for query in queries:
if query[0] == 1:
u = query[1] - 1
v = query[2] - 1
dsu.union(u, v)
connected[u].add(v)
connected[v].add(u)
elif query[0] == 2:
v = query[1] - 1
found = False
for i in range(N):
if i != v and i not in connected[v] and dsu.find(i) != dsu.find(v):
results.append(str(i + 1))
found = True
break
if not found:
results.append('-1')
return "\n".join(results)
import sys
input = sys.stdin.read
data = input().split()
N = int(data[0])
Q = int(data[1])
queries = []
index = 2
for _ in range(Q):
query_type = int(data[index])
if query_type == 1:
u = int(data[index + 1])
v = int(data[index + 2])
queries.append([1, u, v])
index += 3
elif query_type == 2:
v = int(data[index + 1])
queries.append([2, v])
index += 2
print(process_queries(N, Q, queries))