結果

問題 No.92 逃走経路
ユーザー roaris
提出日時 2019-10-28 23:14:24
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 614 bytes
コンパイル時間 162 ms
コンパイル使用メモリ 82,660 KB
実行使用メモリ 578,320 KB
最終ジャッジ日時 2024-09-14 21:21:07
合計ジャッジ時間 12,640 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other MLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

def dfs(cn, depth):
    if depth == K:
        return [cn]
        
    res  = []
    
    for nn, w in adj_list[cn]:
        if d[depth] == w:
            res += dfs(nn, depth+1)
    
    return res

N, M, K = map(int, input().split())
adj_list = [[] for _ in range(N)]

for _ in range(M):
    a, b, c = map(int, input().split())
    adj_list[a-1].append((b-1, c))
    adj_list[b-1].append((a-1, c))

d = list(map(int, input().split()))
ans = set()

for i in range(N):
    ans |= set(dfs(i, 0))

ans = list(sorted(map(lambda x: x+1, list(ans))))

print(len(ans))
print(*ans)
0