結果

問題 No.1393 Median of Walk
コンテスト
ユーザー chineristAC
提出日時 2020-11-12 12:18:29
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,703 bytes
コンパイル時間 172 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 90,496 KB
最終ジャッジ日時 2024-07-19 22:47:18
合計ジャッジ時間 4,647 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 5 TLE * 1 -- * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys

sys.setrecursionlimit(2*10**5)

n,m = map(int,input().split())
edge = [{} for i in range(n)]
Edge = []
for _ in range(m):
    u,v,c = map(int,input().split())
    edge[u-1][v-1] = 1
    Edge.append((u-1,v-1,c))

cost = [[10**18 for j in range(n)] for i in range(n)]
for i in range(n):
    cost[i][i] = 0
for u,v,c in Edge:
    cost[u][v] = 1

for k in range(n):
    for i in range(n):
        for j in range(n):
            if cost[i][j] > cost[i][k] + cost[k][j]:
                cost[i][j] = cost[i][k] + cost[k][j]

Edge.sort(key=lambda x:x[2])
ans  = [[-1 for j in range(n)] for i in range(n)]

def dfs(i,v,w):
    for nv in edge[v]:
        if cost[i][v]!=-10**18:
            if cost[i][nv] > cost[i][v] + edge[v][nv]:
                cost[i][nv] = cost[i][v] + edge[v][nv]
                if cost[i][nv] < -n:
                    cost[i][nv] = -10**18
                if cost[i][nv]<=0 and ans[i][nv]==-1:
                    ans[i][nv] = w
                dfs(i,nv,w)
        else:
            if cost[i][nv]!=-10**18:
                cost[i][nv] = -10**18
                if cost[i][nv]<=0 and ans[i][nv]==-1:
                    ans[i][nv] = w
                dfs(i,nv,w)

for u,v,c in Edge:
    edge[u][v] = -1
    for i in range(n):
        if cost[i][u] != 10**18:
            if cost[i][v] > cost[i][u] + edge[u][v]:
                cost[i][v] = cost[i][u] + edge[u][v]
                if cost[i][v] < -n:
                    cost[i][v] = -10**18
                if ans[i][v] == -1:
                    ans[i][v] = c
                dfs(i,v,c)

Q = int(input())
res = [-1]*Q
for i in range(Q):
    s,t = map(int,input().split())
    res[i] = ans[s-1][t-1]

print(*res,sep="\n")
0