2019年华为实习生第二场笔试第二题是个旅行商问题,虽然只有5个点可以import itertools产生排列遍历5!=120情况求解(当然也可以写个递归自己生成排列),还是查了下动态规划的解法。
原理来自:旅行推销商问题TSP的动态规划解法
实现来自:Python求解tsp问题(动态规划,简单易懂)
实现里做了小小的修改:
def __init__(self,X,start_node):
self.X = X #距离矩阵
self.start_node = start_node #开始的节点
#len(self.X)加多-1
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:
# s改成s-1
su = su + 2**(s-1) # 二进制转换
return su
完整代码:
# tsp问题
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))] #记录处于x节点,未经历M个节点时,矩阵储存x的下一步是M中哪个节点
def trans