結果

問題 No.92 逃走経路
コンテスト
ユーザー けむけむ
提出日時 2021-08-29 11:18:26
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 2,555 ms / 5,000 ms
コード長 1,097 bytes
コンパイル時間 131 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-11-22 04:32:22
合計ジャッジ時間 7,291 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

def tousou(N, M, K, abcs, ds):
    abs = []
    cs = []
    for i in range(M):
        abc = abcs[i]
        abs.append([abc[0], abc[1]])
        cs.append(abc[2])
    koho = []
    for j in range(M):
        if cs[j] == ds[0]:
            koho.append(abs[j][0])
            koho.append(abs[j][1])
    koho = list(set(koho))
    for k in range(1, K):
        d = ds[k]
        koho2 = []
        for l in range(M):
            if cs[l] == d:
                ab = abs[l]
                if ab[0] in koho:
                    koho2.append(ab[1])
                if ab[1] in koho:
                    koho2.append(ab[0])
        koho = list(set(koho2))
    ans1 = len(koho)
    koho = sorted(koho)
    map_koho = list(map(str, koho))
    ans2 = ' '.join(map_koho)
    return [ans1, ans2]

def main():
    N, M, K = map(int, input().split())
    abcs = [0 for x in range(M)]
    for i in  range(M):
        abcs[i] = list(map(int, input().split()))
    ds = list(map(int, input().split()))
    ans = tousou(N, M, K, abcs, ds)
    print(ans[0])
    print(ans[1])

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