結果

問題 No.92 逃走経路
ユーザー ThetaTheta
提出日時 2022-12-06 18:27:17
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 260 ms / 5,000 ms
コード長 1,014 bytes
コンパイル時間 124 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 13,056 KB
最終ジャッジ日時 2024-10-13 07:41:13
合計ジャッジ時間 2,573 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 81 ms
13,056 KB
testcase_01 AC 34 ms
11,520 KB
testcase_02 AC 34 ms
11,392 KB
testcase_03 AC 34 ms
11,392 KB
testcase_04 AC 34 ms
11,392 KB
testcase_05 AC 40 ms
11,648 KB
testcase_06 AC 42 ms
12,160 KB
testcase_07 AC 40 ms
12,288 KB
testcase_08 AC 52 ms
12,160 KB
testcase_09 AC 63 ms
12,416 KB
testcase_10 AC 260 ms
12,160 KB
testcase_11 AC 247 ms
12,160 KB
testcase_12 AC 77 ms
12,672 KB
testcase_13 AC 62 ms
12,416 KB
testcase_14 AC 70 ms
12,416 KB
testcase_15 AC 74 ms
12,672 KB
testcase_16 AC 70 ms
12,672 KB
testcase_17 AC 74 ms
13,056 KB
testcase_18 AC 56 ms
12,288 KB
testcase_19 AC 48 ms
12,160 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque
from typing import Deque


def main():
    N, M, K = map(int, input().split())
    routes = [list(map(int, input().split())) for _ in range(M)]
    routes_cost_dict = defaultdict(lambda: defaultdict(set))
    for route in routes:
        routes_cost_dict[route[2]][route[0]].add(route[1])
        routes_cost_dict[route[2]][route[1]].add(route[0])
    history = list(map(int, input().split()))

    dp_table = [[False for _ in range(N)] for _ in range(K+1)]
    dp_table[0] = [True for _ in range(N)]

    for ctr, cost in enumerate(history, 1):
        for city_idx in range(1, N+1):
            for dest_city_idx in routes_cost_dict[cost][city_idx]:
                dp_table[ctr][dest_city_idx-1] |= dp_table[ctr-1][city_idx-1]

    last_city_candidates = set(
        filter(lambda city_idx: dp_table[K][city_idx], range(N)))
    print(len(last_city_candidates))
    print(*map(lambda num: num+1, sorted(last_city_candidates)))


if __name__ == "__main__":
    main()
0