結果
問題 | No.92 逃走経路 |
ユーザー |
|
提出日時 | 2021-01-17 03:48:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 224 ms / 5,000 ms |
コード長 | 1,177 bytes |
コンパイル時間 | 388 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 89,864 KB |
最終ジャッジ日時 | 2024-11-29 01:34:22 |
合計ジャッジ時間 | 4,662 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#region Header#!/usr/bin/env python3# from typing import *import sysimport ioimport mathimport collectionsimport decimalimport itertoolsfrom queue import PriorityQueueimport bisectimport heapqdef input():return sys.stdin.readline()[:-1]sys.setrecursionlimit(1000000)#endregion# _INPUT = """# paste here...# """# sys.stdin = io.StringIO(_INPUT)def main():N, M, K = map(int, input().split())G = [list() for _ in range(N)]d = dict()for _ in range(M):a, b, c = map(int, input().split())a -= 1b -= 1G[a].append((b, c))G[b].append((a, c))if c not in d:d[c] = set()d[c].add(a)d[c].add(b)D = list(map(int, input().split()))cities = d[D[0]]for i in range(1, K):if not cities:breakcities1 = set()for a in cities:for (b, c) in G[a]:if c == D[i]:cities1.add(b)cities = cities1print(len(cities))print(*sorted([a+1 for a in cities]))if __name__ == '__main__':main()