FR共轭梯度法 matlab

% FR共轭梯度法
function sixge
x0=[1,0]';
[x,val,k]=frcg('fun','gfun',x0)
end

function f=fun(x) 
f=100*(x(1)^2-x(2))^2+(x(1)-1)^2; 
end

function g=gfun(x) 
g=[400*x(1)*(x(1)^2-x(2))+2*(x(1)-1), -200*(x(1)^2-x(2))]'; 
end


function [x,val,k]=frcg(fun,gfun,x0) 
maxk=5000; %最大迭代次数 
rho=0.6;sigma=0.4; 
k=0; epsilon=1e-4; 
n=length(x0); 
while(k<maxk) 
    g=feval(gfun,x0); %计算梯度
    itern=k-(n+1)*floor(k/(n+1)); 
    itern=itern+1; %计算搜索方向 
    if(itern==1) 
        d=-g; 
    else
        beta=(g'*g)/(g0'*g0); 
        d=-g+beta*d0; 
        gd=g'*d; 
        if(gd>=0.0) 
            d=-g; 
        end
    end
    if(norm(g)<epsilon), break; end
    %检验终止条件 
    m=0; mk=0; 
    while(m<15) %Armijo搜索 
        if(feval(fun,x0+rho^m*d)<feval(fun,x0)+sigma*rho^m*g'*d) 
            mk=m; break; 
        end
        m=m+1;
    end
    x0=x0+rho^mk*d; 
    val=feval(fun,x0); 
    g0=g; d0=d; 
    k=k+1;
end
x=x0; 
val=feval(fun,x);
end