結果
問題 | No.92 逃走経路 |
ユーザー |
|
提出日時 | 2024-03-05 23:48:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 101 ms / 5,000 ms |
コード長 | 991 bytes |
コンパイル時間 | 271 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 76,672 KB |
最終ジャッジ日時 | 2024-09-29 18:07:56 |
合計ジャッジ時間 | 2,640 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
import sys import math import bisect from heapq import heapify, heappop, heappush from collections import deque, defaultdict, Counter from functools import lru_cache from itertools import accumulate, combinations, permutations, product sys.setrecursionlimit(1000000) MOD = 10 ** 9 + 7 MOD99 = 998244353 input = lambda: sys.stdin.readline().strip() NI = lambda: int(input()) NMI = lambda: map(int, input().split()) NLI = lambda: list(NMI()) SI = lambda: input() SMI = lambda: input().split() SLI = lambda: list(SMI()) EI = lambda m: [NLI() for _ in range(m)] def main(): N, M, K = NMI() ABC = EI(M) C2I = defaultdict(list) for a, b, c in ABC: C2I[c].append([a, b]) D = NLI() S = set(range(1, N+1)) for d in D: T = set() for a, b in C2I[d]: if a in S: T.add(b) if b in S: T.add(a) S = T print(len(S)) print(*sorted(list(S))) if __name__ == "__main__": main()