旅行商問題的動態規划解法


一個售貨員必須訪問n個城市,這n個城市是一個完全圖,售貨員需要恰好訪問所有城市的一次,並且回到最終的城市。
城市與城市之間有一個旅行費用,售貨員希望旅行費用之和最少。
旅行商問題是np問題,一般可以使用回溯法或者動態規划解決。

class Solution:
    def __init__(self, X, start_node):
        self.X = X
        self.start_node = start_node
        self.array = [[0] * (2 ** (len(self.X) - 1)) for i in range(len(self.X))]

    def transfer(self, sets):
        su = 0
        for s in sets:
            su = su + 2 ** (s - 1)
        return su

    def tsp(self):
        s = self.start_node
        num = len(self.X)
        cities = list(range(num))
        cities.pop(cities.index(s))
        node = s
        return self.solve(node, cities)

    def solve(self, node, future_sets):
        if len(future_sets) == 0:
            return self.X[node][self.start_node]
        d = 9999999
        distance = []
        for i in range(len(future_sets)):
            s_i = future_sets[i]
            copy = future_sets[:]
            copy.pop(i)
            distance.append(self.X[node][s_i] + self.solve(s_i, copy))

        d = min(distance)
        next_one = future_sets[distance.index(d)]
        c = self.transfer(future_sets)
        self.array[node][c] = next_one
        return d


n = int(input())
m = int(input())
D = [[9999999 for j in range(n)] for i in range(n)]
for i in range(m):
    a, b, t = list(map(float, input().split()))
    a,b = int(a),int(b)
    D[a][b] = t
    D[b][a] = t

S = Solution(D, 0)
res = int(S.tsp())
if res >= 9999999:
    print(-1)
else:
    print(res)

參考:
漫畫:什么是旅行商問題?
旅行商問題動態規划解法(python版)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM