結果

問題 No.1477 Lamps on Graph
ユーザー tobusakanatobusakana
提出日時 2022-11-27 14:34:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 448 ms / 2,000 ms
コード長 1,259 bytes
コンパイル時間 304 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 103,548 KB
最終ジャッジ日時 2024-10-04 02:48:07
合計ジャッジ時間 11,616 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 番号が小さい頂点から大きい頂点に辺を貼る有向グラフを構成する
# 点灯している頂点で且つ番号が小さいものは、自分を消すしかないので、それを処理
# heapqに点灯している頂点の番号を入れていき、それを順に処理すればOK
# ループすることはないので、必ず全て消すことが出来る

import sys
readline = sys.stdin.readline
N,M = map(int,readline().split())
A = list(map(int,readline().split()))
G = [[] for i in range(N)]
for _ in range(M):
    u,v = map(int,readline().split())
    u -= 1
    v -= 1
    if A[u] < A[v]:
        G[u].append(v)
    if A[u] > A[v]:
        G[v].append(u)
        
import heapq as hq
q = []
K = int(readline())
B = list(map(int,readline().split()))

lamp_on = [False] * N
for b in B:
    lamp_on[b - 1] = True
    hq.heappush(q, (A[b - 1], b - 1)) # 持ってる数値,頂点番号
    
ans = []
while q:
    num, v = hq.heappop(q)
    if not lamp_on[v]: # 既に消えていれば対象外
        continue
    lamp_on[v] = False
    ans.append(v + 1)
    for child in G[v]:
        lamp_on[child] ^= True
        if lamp_on[child]:
            hq.heappush(q, (A[child], child))

print(len(ans))
for a in ans:
    print(a)
0