結果

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

ソースコード

diff # 

import sys
input = sys.stdin.readline
import heapq

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

EDGE=[dict() for i in range(2*N+1)]

for i in range(M):
    u,v,c,d=map(int,input().split())
    EDGE[u][v]=[c,1]
    EDGE[u][N+v]=[d,1]
    EDGE[u][N+u]=[0,float("inf")]
    EDGE[N+u][u]=[0,float("inf")]
    EDGE[N+v][u]=[d,0]

    EDGE[v][u]=[c,1]
    EDGE[v][N+u]=[d,1]
    EDGE[v][N+v]=[0,float("inf")]
    EDGE[N+v][v]=[0,float("inf")]
    EDGE[N+u][v]=[d,0]


start=1
goal=N

BACK=[-1]*(2*N+1)
LA=0

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

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

    LA+=ANS[goal]
 
    NOW=goal
    while NOW!=start:
        fr=BACK[NOW]
        EDGE[fr][NOW][1]-=1
        EDGE[NOW][fr][1]+=1

        NOW=fr

print(LA)
0