Dijkstra算法的麻烦及其性能

内存限制: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)
wowkrad 回答:Dijkstra算法的麻烦及其性能

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3159482.html

大家都在问