結果

問題 No.1301 Strange Graph Shortest Path
ユーザー titia
提出日時 2020-11-28 08:38:39
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,643 bytes
コンパイル時間 505 ms
コンパイル使用メモリ 82,408 KB
実行使用メモリ 254,920 KB
最終ジャッジ日時 2024-09-12 22:10:25
合計ジャッジ時間 43,004 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 32 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

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)]
P=[0]*(N+1)

for fl in range(2):
    ANS=[float("inf")]*(N+1)
    Q=[start]
    ANS[start]=0
 
    while Q:
        x = heapq.heappop(Q)
        time = x>>20
        fr = x- (time<<20)
        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]<<20)+to)

    P=[P[i]+ANS[i] for i in range(N+1)]
 
    NOW=goal
    while NOW!=start:
        fr,ind=BACK[NOW]
        EDGELIST[ind][2]-=1
        EDGELIST[ind^1][2]+=1

        NOW=fr

    for i in range(N+1):
        for j in range(len(EDGE[i])):
            EDGE[i][j][1]+=P[EDGE[i][j][0]]-P[i]

print(ANS[goal])
0