結果

問題 No.92 逃走経路
ユーザー roaris
提出日時 2019-10-28 23:18:15
言語 PyPy3
(7.0.0)
結果
TLE  
実行時間 -
コード長 643 Byte
コンパイル時間 240 ms
使用メモリ 257,624 KB
最終ジャッジ日時 2019-10-28 23:18:28

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
challenge01.txt TLE -
sample01.txt -- -
sample02.txt -- -
test01.txt -- -
test02.txt -- -
test03.txt -- -
test04.txt -- -
test05.txt -- -
test06.txt -- -
test07.txt -- -
test08.txt -- -
test09.txt -- -
test10.txt -- -
test11.txt -- -
test12.txt -- -
test13.txt -- -
test14.txt -- -
test15.txt -- -
test16.txt -- -
test17.txt -- -
テストケース一括ダウンロード

ソースコード

diff #
import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**8)

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