結果

問題 No.807 umg tours
ユーザー tomarint2
提出日時 2019-03-22 23:22:54
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,966 bytes
コンパイル時間 345 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 852,480 KB
最終ジャッジ日時 2024-09-19 06:57:48
合計ジャッジ時間 5,108 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 11 MLE * 1 -- * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import copy
import collections

N,M=map(int,sys.stdin.readline().rstrip().split())
d=[[0 for j in range(N+1)] for i in range(N+1)]
for i in range(M):
    a,b,c=map(int,sys.stdin.readline().rstrip().split())
    d[a][b]=d[b][a]=c
inf=1<<60

#現在地、ゴール、コスト、チケット利用回数、履歴
def solve0(location, goal, cost, ticket, history):
    cost2 = inf
    #print('solve', location, goal, cost, ticket)
    history2 = copy.copy(history)
    history2.add(location)
    if location==goal:
        return cost
    for i in range(1,N+1):
        if d[location][i]==0:
            continue
        if i in history2:
            continue
        cost2=min(cost2,solve(i,goal,cost+d[location][i],ticket,history2))
        if ticket==0:
            cost2=min(cost2,solve(i,goal,cost,1,history2))
    return cost2

def solve(location, goal, cost, ticket, history):
    q = collections.deque()
    q.append((location, cost, ticket, history))
    cost2 = inf
    lcost=[inf for i in range(N+1)]
    try:
        while True:
            location, cost, ticket, history = q.popleft()
            #print('solve', location, cost, ticket, history)
            if lcost[location]<location:
                continue
            if location==goal:
                cost2=min(cost2,cost)
                continue
            lcost[location]=location
            if cost >= cost2:
                continue
            history2 = copy.copy(history)
            history2.add(location)
            for i in range(1,N+1):
                if d[location][i]==0:
                    continue
                if i in history2:
                    continue
                if ticket==0:
                    q.append((i, cost, 1, history2))
                q.append((i, cost+d[location][i], ticket, history2))
    except IndexError:
        pass
    return cost2

for i in range(1,N+1):
    ans=solve(1, i, 0, 0, set()) + solve(1, i, 0, 1, set())
    print(ans)
0