travel.m

来自「简单蚁群算法解决tsp问题 一共有48城市 花费7分钟」· M 代码 · 共 31 行

function [y,val]=travel(distance,tao,alpha,beta) % 某只蚂蚁找到的某条路径 
[m,n]=size(distance); 
p=fix(m*rand)+1; 
val=0; % 初始路径长度设为 0 
tabuk=[p]; % 假设该蚂蚁都是从第 p 个城市出发的 
for i=1:m-1 
np=tabuk(length(tabuk)); % 蚂蚁当前所在的城市号 
p_sum=0; 
for j=1:m 
if isin(j,tabuk) 
continue; 
else 
ada=1/distance(np,j); 
p_sum=p_sum+tao(np,j)^alpha*ada^beta; 
end 
end 
cp=zeros(1,m); % 转移概率 
for j=1:m 
if isin(j,tabuk) 
continue; 
else 
ada=1/distance(np,j); 
cp(j)=tao(np,j)^alpha*ada^beta/p_sum; 
end 
end 
NextCity=pchoice(cp); 
tabuk=[tabuk,NextCity]; 
val=val+distance(np,NextCity); 
end 
y=tabuk;