結果

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

ソースコード

diff # 

n,m = map(int,input().split())
edge = [[] for i in range(n)]
edge_cost = {}
Edge = []
for _ in range(m):
    u,v,c = map(int,input().split())
    edge[u-1].append(v-1)
    edge_cost[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)]
dfs = [[] for i in range(n)]

for u,v,w in Edge:
    edge_cost[u,v] = -1
    for i in range(n):
        if cost[i][u] != 10**18:
            if cost[i][v] > cost[i][u] + edge_cost[u,v]:
                cost[i][v] = cost[i][u] + edge_cost[u,v]
                if cost[i][v] < -n:
                    cost[i][v] = -10**18
                if cost[i][v]<=0 and ans[i][v] == -1:
                    ans[i][v] = w
                dfs[i].append([v,0])

            while dfs[i]:
                v,idx = dfs[i][-1]
                if idx!=len(edge[v]):
                    dfs[i][-1][1] += 1
                    nv = edge[v][idx]
                    if cost[i][v]!=-10**18:
                        if cost[i][nv] > cost[i][v] + edge_cost[v,nv]:
                            cost[i][nv] = cost[i][v] + edge_cost[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].append([nv,0])
                    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].append([nv,0])
                else:
                    dfs[i].pop()

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