用matlab2021b写一个蚁群算法

好的，下面是一个简单的蚁群算法的 Matlab 代码实现。

% 蚁群算法
% 目标函数为f(x)=x^2
% 求解[0,5]范围内的最小值

% 参数设置
nAnt = 20; % 蚂蚁数量
nIter = 100; % 迭代次数
alpha = 1; % 信息素重要程度因子
beta = 5; % 启发式因子
rho = 0.1; % 信息素挥发因子
Q = 1; % 常数因子
L = 5; % 信息素初始值
xMin = 0; % 搜索区间左端点
xMax = 5; % 搜索区间右端点

% 初始化
x = xMin + (xMax - xMin) * rand(1, nAnt); % 每只蚂蚁的位置
t = ones(1, nAnt) * L; % 每只蚂蚁所在位置的信息素
xBest = x(1); % 全局最优解
fBest = xBest^2; % 全局最优解对应的函数值

% 迭代
for iter = 1:nIter
    % 计算每只蚂蚁的概率分布
    p = t.^alpha .* (1 ./ (x.^2 + eps)).^beta;
    p = p / sum(p);
    
    % 蚂蚁移动
    xNew = zeros(1, nAnt);
    for k = 1:nAnt
        % 选择下一个位置
        cumProb = cumsum(p);
        r = rand;
        j = find(cumProb >= r, 1, 'first');
        xNew(k) = x(j);
        
        % 更新信息素
        t(j) = (1 - rho) * t(j) + Q / (xNew(k)^2 + eps);
        
        % 更新全局最优解
        fNew = xNew(k)^2;
        if fNew < fBest
            xBest = xNew(k);
            fBest = fNew;
        end
    end
    
    % 更新蚂蚁位置
    x = xNew;
end

disp(['最优解为：', num2str(xBest)]);
disp(['最优解对应的函数值为：', num2str(fBest)]);

代码中的目标函数为 $f(x)=x^2$，在 $[0,5]$ 区间内寻找最小值。参数设置包括蚂蚁数量、迭代次数、信息素重要程度因子、启发式因子、信息素挥发因子、常数因子等。在迭代过程中，先计算每只蚂蚁的概率分布，然后根据概率分布选择下一个位置，更新信息素和全局最优解。最后输出最优解和最优解对应的函数值。

注意，该代码实现的是基本的蚁群算法，实际应用中可能需要根据具体问题进行改进和优化。