内存限制:512 MB,时间限制:4s
如果您的汽车油箱= k,此算法将显示是否有可能从v骑乘到u。如果不可能,算法会搜索最佳的加注量(加注水箱装满后)。但是我的代码与时间限制不符。请帮我加电
输入:
(在一行中):k-坦克,n-城市数量,m-城市之间的道路数量,u-起始城市,v-终止城市
(接下来的m行):q,p,-与道路m和r连接的城市数-这条道路的长度
(下一行):l-笔芯数量
(一行):具有笔芯的城市
输出:最小笔芯数。
示例:
输入:3 3 3 1 3 1 2 3 1 3 4 2 3 3 2 2 3 输出:1
输入:3 3 3 1 3 1 2 2 1 3 4 2 3 2 0 输出:-1
输入:3 3 3 1 3 1 2 2 1 3 4 2 3 1 0 输出:0
def main: k,n,m,u,v = map(int,input()。split())
w = [[0] * n for i in range(n)]
for j in range(m):
p,q,r = map(int,input().split())
w[p - 1][q - 1] = r
w[q - 1][p - 1] = r
if (min(w[u-1]) > k) or (min(w[v-1]) > k):
print("-1")
exit(0)
f = int(input())
if f > 0:
refills = list(map(int,input().split()))
availableRefills = [0] * n
def dejkstr(source,dest):
source -= 1
dest -= 1
v = 0
visited = [False] * n
D = [10000000000] * n
visited[source] = True
for i in range(n):
if w[source][i] != 0:
D[i] = w[source][i]
D[source] = 0
for i in range(n):
minx = 10000000000
for j in range(n):
if (D[j] < minx) and (D[j] != 0) and (not visited[j]):
minx = D[j]
v = j
for j in range(n):
if (not visited[j]) and (D[j] > D[v] + w[v][j]) and (w[v][j] != 0):
D[j] = D[v] + w[v][j]
source = v
visited[v] = True
return D[dest]
fuelStations = 0
while (u != v) and (fuelStations < f):
kmNeeded = dejkstr(u,v)
if k >= kmNeeded:
print(fuelStations)
exit(0)
else:
if f == 0:
print("-1")
exit(0)
else:
j = 0
for i in range(f):
tarFuelCity = refills[i]
kmNeeded = dejkstr(u,tarFuelCity)
if (kmNeeded <= k) and (tarFuelCity != u):
availableRefills[j] = tarFuelCity
j += 1
bestRefillKM = 10000000000
bestRefill = 0
for i in range(f):
if availableRefills[i] > 0:
overallKM = dejkstr(u,availableRefills[i])
overallKM = overallKM + dejkstr(availableRefills[i],v)
if overallKM < bestRefillKM:
bestRefillKM = overallKM
bestRefill = availableRefills[i]
u = bestRefill
fuelStations += 1
for i in range(len(availableRefills)):
availableRefills[i] = 0
if u == v:
print(fuelStations)
else:
print("-1")
exit(0)