智能优化算法：粒子群算法2——二元函数极值问题

% 二元函数最值问题

close all
clear
clc

N = 100;            % 粒子数
D = 2;              % 维度
T = 200;            % 最大迭代次数

c1 = 1.5;           % 学习因子1
c2 = 1.5;
Wmax = 0.8;
Wmin = 0.4;

Xmax = 4;
Xmin = -4;
Vmax = 0.5;
Vmin = -0.5;

x = rand(N,D) * (Xmax-Xmin) + Xmin;
v = rand(N,D) * (Vmax-Vmin) + Vmin;

p = x;
pbest = ones(N,1);
for i = 1:N
    pbest(i) = f(x(i,:));
end

g = ones(1,D);
gbest = inf;
for i = 1:N
    if(pbest(i) < gbest)
        g = p(i,:);
        gbest = pbest(i);
    end
end

gb = ones(1,T);

% 开始迭代
for i = 1:T
    for j = 1:N
        if f(x(j,:)) < pbest(j)
            p(j,:) = x(j,:);
            pbest(j) = f(x(j,:));
        end
        if pbest(j) < gbest
            g = p(j,:);
            gbest = pbest(j);
        end
        w = Wmax - (Wmax - Wmin)*i/T;
        v(j,:) = w * v(j,:) + c1 * rand * (p(j,:) - x(j,:)) + c2 * rand * (g - x(j,:));
        x(j,:) = x(j,:) + v(j,:);

        for ii = 1:D
            if (v(j,ii) > Vmax) || (v(j,ii) < Vmin)
                v(j,ii) = rand * (Vmax-Vmin) + Vmin;
            end


            if (x(j,ii) > Xmax) || (x(j,ii) < Xmin)
                x(j,ii) = rand * (Xmax-Xmin) + Xmin;
            end
        end
    end
    gb(i) = gbest;
end % 结束迭代

Wfig = 400;     % 绘图宽度（300，对应生成图为8cm宽，复合论文用图）
Hfig = 350;     % 绘图高度
figure('name','迭代曲线','color','w');
set(gcf,'position',[768-Wfig/2,452-Hfig/2,Wfig,Hfig]);
plot(gb,'b','linewidth',1);
grid on
set(gca,'FontName','Times New Roman','FontSize',12);
xlabel('迭代次数','FontName','宋体');
ylabel('最优解','FontName','宋体');
xlim([1,T])

Nn = 100;
x1 = linspace(Xmin,Xmax,Nn);
x2 = linspace(Xmin,Xmax,Nn);
[X1,X2] = meshgrid(x1,x2);
Y = zeros(Nn,Nn);
for i = 1:Nn
    for j = 1:Nn
        Y(i,j) = f([X1(i,j),X2(i,j)]);
    end
end

figure('name','函数图像','color','w')
set(gcf,'position',[768-Wfig/2,452-Hfig/2,Wfig,Hfig]);
s = surf(X1,X2,Y);
shading interp;
colormap(jet);

hold on
set(gca,'FontName','Times New Roman','FontSize',12);
plot3(g(1,1),g(1,2),gb(end),'ro')
box on

function y = f(x)
y = 3 * cos(x(1,1)*x(1,2))+x(1,1) + x(1,2)^2;
end