結果

問題 No.1301 Strange Graph Shortest Path
コンテスト
ユーザー titia
提出日時 2020-11-28 07:19:48
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,448 bytes
コンパイル時間 488 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 232,064 KB
最終ジャッジ日時 2024-09-12 22:05:40
合計ジャッジ時間 34,471 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
import heapq

N,M=map(int,input().split())

EDGE=[[] for i in range(N+1)]
EDGELIST=[]

eind=0
for i in range(M):
    u,v,c,d=map(int,input().split())
    X=[v,c,1,eind]
    EDGE[u].append(X)
    EDGELIST.append(X)
    eind+=1

    X=[u,c,0,eind]
    EDGE[v].append(X)
    EDGELIST.append(X)
    eind+=1

    X=[u,c,1,eind]
    EDGE[v].append(X)
    EDGELIST.append(X)
    eind+=1

    X=[v,c,0,eind]
    EDGE[u].append(X)
    EDGELIST.append(X)
    eind+=1


    X=[v,d,1,eind]
    EDGE[u].append(X)
    EDGELIST.append(X)
    eind+=1

    X=[u,d,0,eind]
    EDGE[v].append(X)
    EDGELIST.append(X)
    eind+=1

    X=[u,d,1,eind]
    EDGE[v].append(X)
    EDGELIST.append(X)
    eind+=1

    X=[v,d,0,eind]
    EDGE[u].append(X)
    EDGELIST.append(X)
    eind+=1


start=1
goal=N

BACK=[[-1,-1] for i in range(N+1)]
LA=0

for fl in range(2):
    ANS=[float("inf")]*(N+1)
    Q=[(0,start)]
    ANS[start]=0
 
    while Q:
        time,fr = heapq.heappop(Q)
        if time > ANS[fr]:
            continue

        for to,cost,vol,ind in EDGE[fr]:
            
            if vol>0 and ANS[to]>ANS[fr]+cost:
                
                ANS[to]=ANS[fr]+cost
                BACK[to]=[fr,ind]
                heapq.heappush(Q,(ANS[to],to))

    LA+=ANS[goal]
 
    NOW=goal
    while NOW!=start:
        fr,ind=BACK[NOW]
        EDGELIST[ind][2]-=1
        EDGELIST[ind^1][2]+=1

        NOW=fr

print(LA)
0