結果

問題 No.3158 Collect Stamps
コンテスト
ユーザー tkykwtnb
提出日時 2025-05-23 20:08:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 104 ms / 2,000 ms
コード長 786 bytes
コンパイル時間 536 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 86,388 KB
最終ジャッジ日時 2025-05-23 20:08:54
合計ジャッジ時間 3,626 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
N,M,K=map(int,input().split())
A=set(map(lambda x:int(x)-1,input().split()))
T=[list(map(int,input().split())) for _ in range(N)]
G=[[] for _ in range(N)]
for i in range(N):
    for j in range(i+1,N):
        if i==j:continue
        G[i].append(j)
        G[j].append(i)
ans=1<<60
for s in range(N):
    vis=[0 for _ in range(N)]
    Q=deque([(s,0,0,tuple(vis))])
    while Q:
        now,cnt,t,vis=Q.popleft()
        vis=list(vis)
        if vis[now]==0:
            cnt+=1
        vis[now]=1
        if cnt>=M and now in A:
            ans=min(ans,t)
            continue
        for nxt in G[now]:
            if cnt<M and vis[nxt]:continue
            if t+T[now][nxt]>ans:continue
            Q.append((nxt,cnt,t+T[now][nxt],tuple(vis)))
print(ans)
0