"""模拟退火旅行商"""
import random
import numpy as np
import math
import time
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
location = np.loadtxt('city_location.txt') # 加载数据
'''一些关键参数'''
num_city = 30 # 城市总数
initial_t = 100 # 初始温度
lowest_t = 0.001 # 最低温度
iteration = 10000 # 设置迭代次数
'''输入两个城市之间的距离'''
def distance_mat():
distance = [] # 初始化距离数组
for i in range(30):
distance_each = [] # 初始化每个城市到其他城市的距离
for j in range(30):
dis = math.sqrt(pow(location[i][0] - location[j][0], 2) +
pow(location[i][1] - location[j][1], 2)) # 计算距离
distance_each.append(dis) # 赋值到distance_each数组
distance.append(distance_each) # 按列添加
return distance
'''计算所有路径对应的距离'''
def cal_newpath(distance, path):
# 此时传入的path是一条随机生成的路径
dis = 0
for j in range(num_city - 1): # 只遍历0到28个城市，是因为第29个城市的下一个是起点，这样才是TSP问题，形成闭环
dis = distance[path[j]][path[j + 1]] + dis # 这条路径上经过的两两个点的距离之和即为这条路径的长度
dis = dis + distance[path[29]][path[0]] # 计算的距离之和再加上第28个城市回到起点的距离
return dis
'''点对点的距离矩阵'''
distance = distance_mat()
'''初始路径'''
path = list(range(30)) # 生成0到29的列表，即城市索引
'''初始距离'''
dis = cal_newpath(distance, path) # 先计算初始的距离，这样在模拟退火的时候才可以开始比较
'''初始温度'''
t_current = initial_t
'''灵敏度分析'''
sensibility = []
start_time = time.time() # 开始计时
'''模拟退火'''
while t_current > lowest_t: # 外层循环：改变温度
count_m = 0 # M的计数
count_iter = 0 # 迭代次数计数
while count_iter < iteration: # 内层循环:连续多次不接受新的状态则跳出内循环
i = 0
j = 0
while i == j: # 防止随机了同一城市
i = random.randint(1, 29)
j = random.randint(1, 29)
path_new = path.copy()
path_new[i], path_new[j] = path_new[j], path_new[i] # 任意交换两个城市的位置,产生新的路径组合
dis_new = cal_newpath(distance, path_new) # 计算新路径的距离
dis_delta = dis_new - dis # 计算新距离和旧距离的差值
rand = random.random() # 生成一个0到1的浮点随机数
exp_d = math.exp(-dis_delta / t_current) # Metropolis准则
'''是否接受新解的过程'''
if dis_delta < 0: # 如果新距离小于旧距离，则直接接受
path = path_new
dis = dis_new
elif exp_d > rand: # 如果新距离大于旧距离，用Metropolis准则判断是否接受
path = path_new
dis = dis_new
else: # 不接受新解
count_m = count_m + 1
count_iter = count_iter + 1 # 迭代次数加1
sensibility.append(dis)
t_current = 0.99 * t_current # 改变温度
end_time = time.time()
elapsed_time = end_time - start_time
# 绘制最短路径的坐标图
x_coords = [location[i][0] for i in path]
y_coords = [location[i][1] for i in path]
# 添加起点到终点的连线
x_coords.append(x_coords[0])
y_coords.append(y_coords[0])
plt.figure(1)
plt.plot(x_coords, y_coords, marker='o', linestyle='-')
plt.title('最短路径坐标图')
plt.xlabel('X 坐标')
plt.ylabel('Y 坐标')
plt.grid(True)
plt.show()
plt.figure(2)
plt.plot(sensibility,marker='.',color='r',markersize=3)
plt.title('最短路径坐标图')
plt.xlabel('迭代次数')
plt.ylabel('最短距离')
plt.grid(True)
plt.show()
'''输出结果'''
print('最短距离：', dis)
print('最短路径：', path)
print('运行时间：', elapsed_time, '秒')最短距离： 424.69177537685437
最短路径： [0, 5, 4, 3, 12, 11, 29, 22, 21, 16, 15, 28, 27, 26, 25, 24, 23, 14, 13, 7, 9, 20, 19, 18, 6, 10, 8, 2, 17, 1]
运行时间： 43.86513066291809 秒
