algorithm/leetcode-master
1
0
Fork 0
mirror of https://github.com/youngyangyang04/leetcode-master.git synced 2025-07-06 07:06:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: error saving the edges

Browse Source
This commit is contained in:
kyshen
2024-09-06 23:59:51 +08:00
parent e391a24729
commit e24c307c07
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
problems/kamacoder/0094.城市间货物运输I-SPFA.md
View File

@ -430,10 +430,10 @@ import collections
def main():
n, m = map(int, input().strip().split())
edges = []
edges = [[] for _ in range(n + 1)]
for _ in range(m):
src, dest, weight = map(int, input().strip().split())
edges.append([src, dest, weight])
edges[src].append([dest, weight])
minDist = [float("inf")] * (n + 1)
minDist[1] = 0
@ -443,18 +443,13 @@ def main():
while que:
cur = que.popleft()
visited[cur] = True
updated = False
for src, dest, weight in edges:
if minDist[src] != float("inf") and minDist[src] + weight < minDist[dest]:
minDist[dest] = minDist[src] + weight
updated = True
visited[cur] = False
for dest, weight in edges[cur]:
if minDist[cur] != float("inf") and minDist[cur] + weight < minDist[dest]:
minDist[dest] = minDist[cur] + weight
if visited[dest] == False:
que.append(dest)
visited[dest] = True
if not updated:
break
if minDist[-1] == float("inf"):
return "unconnected"