結果

問題 No.1477 Lamps on Graph
コンテスト
ユーザー ああいい
提出日時 2021-12-27 15:17:13
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,054 ms / 2,000 ms
コード長 696 bytes
コンパイル時間 121 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 37,608 KB
最終ジャッジ日時 2024-09-25 08:46:23
合計ジャッジ時間 19,798 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M = map(int,input().split())
A = list(map(int,input().split()))
G = [[] for _ in range(N+1)]
for _ in range(M):
    u,v = map(int,input().split())
    if A[u-1] < A[v-1]:
        G[u].append(v)
    elif A[u-1] > A[v-1]:
        G[v].append(u)
K = int(input())
B = list(map(int,input().split()))
memo = [False] * (N + 1)
for b in B:
    memo[b] = True
import heapq
q = [(A[b-1],b) for b in B]
heapq.heapify(q)
ans = []
while (len(q)):
    a,b = heapq.heappop(q)
    if memo[b] == False:
        continue
    ans.append(b)
    memo[b] = False
    for v in G[b]:
        memo[v] = not memo[v]
        if memo[v]:
            heapq.heappush(q,(A[v-1],v))
print(len(ans))
for a in ans:
    print(a)
0