結果
問題 |
No.1477 Lamps on Graph
|
ユーザー |
![]() |
提出日時 | 2022-11-29 22:36:09 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 753 bytes |
コンパイル時間 | 450 ms |
コンパイル使用メモリ | 82,244 KB |
実行使用メモリ | 98,592 KB |
最終ジャッジ日時 | 2024-10-07 08:46:32 |
合計ジャッジ時間 | 10,657 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 16 TLE * 1 -- * 21 |
ソースコード
import heapq import sys input = sys.stdin.readline n, m = map(int, input().split()) a = list(map(int, input().split())) adj = [[] for _ in range(n)] for _ in range(m): u, v = map(int, input().split()) u -= 1 v -= 1 if a[u] < a[v]: adj[u].append(v) if a[v] < a[u]: adj[v].append(u) _ = int(input()) b = list(map(int, input().split())) onoff = [(i + 1 in b) for i in range(n)] hq = [(a[i - 1], i - 1) for i in b] heapq.heapify(hq) ans = [] while hq: x, y = heapq.heappop(hq) if not onoff[y]: continue ans.append(y + 1) onoff[y] = False for v in adj[y]: onoff[v] = not onoff[v] if onoff[v]: heapq.heappush(hq, (a[v], v)) print(len(ans)) print(*ans, sep='\n')