結果

問題 No.2319 Friends+
コンテスト
ユーザー Shirotsume
提出日時 2023-05-26 22:42:56
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,112 bytes
コンパイル時間 846 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 200,448 KB
最終ジャッジ日時 2024-12-25 09:04:32
合計ジャッジ時間 23,945 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 44 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque, Counter
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353

n, m = mi()

nowworld = [v - 1 for v in li()] #n人の今いるワールド

worldlist = [set() for _ in range(n)]

for i in range(n):
    worldlist[nowworld[i]].add(i)
graph = [set() for _ in range(n)]

for _ in range(m):
    u, v = mi()
    u -= 1; v -= 1
    graph[u].add(v)
    graph[v].add(u)

def common(x, y):
    if len(graph[x]) < len(worldlist[nowworld[y]]):
        for v in graph[x]:
            if v in worldlist[nowworld[y]]:
                return True
        return False
    else:
        for v in worldlist[nowworld[y]]:
            if v in graph[x]:
                return True
        return False
for _ in range(ii()):
    x, y = mi()
    x -= 1; y -= 1
    if nowworld[x] != nowworld[y] and common(x, y):
        worldlist[nowworld[x]].discard(x)
        nowworld[x] = nowworld[y]
        worldlist[nowworld[x]].add(x)
        print('Yes')
    else:
        print('No')
0