結果

問題 No.92 逃走経路
ユーザー wgrape
提出日時 2025-03-13 16:59:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 81 ms / 5,000 ms
コード長 706 bytes
コンパイル時間 440 ms
コンパイル使用メモリ 82,352 KB
実行使用メモリ 72,488 KB
最終ジャッジ日時 2025-03-13 16:59:39
合計ジャッジ時間 2,919 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

N,M,K = map(int,input().split())
G = [[] for i in range(N)]
for _ in range(M):
    a,b,c = map(int,input().split())
    G[a - 1].append([b - 1, c])
    G[b - 1].append([a - 1, c])
    
D = list(map(int,input().split()))

dp = [[False] * N for i in range(K + 1)]
# dp[k][i] = 次に通る道路の指示がk番目で、i番目にいる
for i in range(N):
    dp[0][i] = True
    
for k in range(K):
    for i in range(N):
        if not dp[k][i]:
            continue
        for child, c in G[i]:
            if D[k] != c:
                continue
            dp[k + 1][child] = True
            
ans = []
for i in range(N):
    if dp[-1][i]:
        ans.append(i + 1)
        
print(len(ans))
print(*ans)
0