結果

問題 No.2569 はじめてのおつかいHard
コンテスト
ユーザー hirayuu_yc
提出日時 2023-12-02 15:34:39
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,686 bytes
コンパイル時間 380 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 329,272 KB
最終ジャッジ日時 2024-09-26 18:55:35
合計ジャッジ時間 11,668 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other AC * 5 TLE * 2 -- * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Graph():
    def __init__(self,size,directed=False):
        self.dir=directed
        self.size=size
        self.gr=[[] for i in range(size)]
        self.edges=[]
    
    def add_edge(self,u,v,status={}):
        self.gr[u].append(self.Edge(u,v,status))
        if not(self.dir):
            self.gr[v].append(self.Edge(v,u,status))
        self.edges.append(self.Edge(u,v,status))

    def node(self,v):
        return self.gr[v]

    def __getitem__(self,v):
        return self.gr[v]
    
    class Edge():
        def __init__(self,st,to,status):
            self.st=st
            self.to=to
            self.status=status
        
        def __getitem__(self,val):
            return self.status[val]
from math import inf
from heapq import heappop,heappush

def dijkstra(graph,start,weight):
    dist=[inf for i in range(graph.size)]
    used=[False for i in range(graph.size)]
    dist[start]=0
    vert=[(0,start)]
    while len(vert)>0:
        dis,pos=heappop(vert)
        if used[pos]:
            continue
        used[pos]=True
        for i in graph.node(pos):
            if dis+weight(i)<dist[i.to]:
                heappush(vert,(dis+weight(i),i.to))
                dist[i.to]=dis+weight(i)
    return dist

N,M=map(int,input().split())
gr=Graph(N,directed=True)
rev=Graph(N,directed=True)
for i in range(M):
    u,v,t=map(int,input().split())
    gr.add_edge(u-1,v-1,{"w":t})
    rev.add_edge(v-1,u-1,{"w":t})
w=lambda x:x["w"]
d1=dijkstra(gr,N-2,w)
d2=dijkstra(gr,N-1,w)
d3=dijkstra(rev,N-2,w)
d4=dijkstra(rev,N-1,w)
for i in range(N-2):
    ans=min(d3[i]+d1[N-1]+d2[i],d4[i]+d2[N-2]+d1[i])
    if ans==inf:
        print(-1)
    else:
        print(ans)
0