Python TSP旅行商问题 暴力求解

原文链接: Python TSP旅行商问题 暴力求解

上一篇: Python 实现 遗传算法

下一篇: HDU 5067 简单tsp

问题概述

旅行商问题，即TSP问题（Traveling Salesman Problem）是数学领域中著名问题之一。假设有一个旅行商人要拜访N个城市，他必须选择所要走的路径，路径的限制是每个城市只能拜访一次，而且最后要回到原来出发的城市。路径的选择目标是要求得的路径路程为所有路径之中的最小值。TSP问题是一个 NPC 问题。

如果存在最短路径

a->b->c->d

则按照顺序即可找到其他路径如

b->c->d->a

c->d->a->b

d->a->b->c

但是在浮点数的情况下，有时候会有微小的误差导致最后结果略有不同

暴力求解的时间复杂度是O(n!)

随机生成dis数组并求解，每次结果都是不一样的。。。

import itertools
import json
import random

# 303.81999999999994
dis = [
    [
        random.randint(1, 10000) if j != i else 0
        for j in range(9)
    ]
    for i in range(9)
]
 
 

def getDis(path):
    d = sum([dis[path[i]][path[i + 1]] for i in range(len(path) - 1)] + [dis[path[-1]][path[0]]])
    return d


min_dis = 1 << 32
min_path = []
all_min_dis = [1 << 32 for _ in range(len(dis))]
for path in itertools.permutations(range(len(dis))):
   
    all_min_dis[path[0]] = min(all_min_dis[path[0]], getDis(path))

    if min_dis > getDis(path):
        min_dis = getDis(path)
        min_path = []
        min_path.append(path)
    elif min_dis == getDis(path):
        min_path.append(path)

for p in min_path:
    print(p, getDis(p))
print(min_dis)
for index, i in enumerate(all_min_dis):
    print(index, i)

(0, 8, 2, 3, 4, 1, 5, 6, 7) 8937
(1, 5, 6, 7, 0, 8, 2, 3, 4) 8937
(2, 3, 4, 1, 5, 6, 7, 0, 8) 8937
(3, 4, 1, 5, 6, 7, 0, 8, 2) 8937
(4, 1, 5, 6, 7, 0, 8, 2, 3) 8937
(5, 6, 7, 0, 8, 2, 3, 4, 1) 8937
(6, 7, 0, 8, 2, 3, 4, 1, 5) 8937
(7, 0, 8, 2, 3, 4, 1, 5, 6) 8937
(8, 2, 3, 4, 1, 5, 6, 7, 0) 8937
8937
0 8937
1 8937
2 8937
3 8937
4 8937
5 8937
6 8937
7 8937
8 8937