bgs.m

来自「Matlab语言常用算法程序集.rar 里面有很多有用的Matlab数值运」· M 代码 · 共 43 行

function [x,N]= BGS(A,b,x0,d,eps,M)
if nargin==4
    eps= 1.0e-6;
    M  = 10000;
elseif nargin<4
    error
    return
elseif nargin ==5
    M  = 10000;
end  

NS = size(A);
n = NS(1,1);
bnum = length(d);
bs = ones(bnum,1);
for i=1:(bnum-1)
    bs(i+1,1)=sum(d(1:i))+1;
    %获得对角线上每个分块矩阵元素索引的起始值
end

DB = zeros(n,n);       
for i=1:bnum
    endb = bs(i,1)+d(i,1)-1;
    DB(bs(i,1):endb,bs(i,1):endb)=A(bs(i,1):endb,bs(i,1):endb);
    %求A的对角分块矩阵
end
LB = -tril(A-DB);      %求A的下三角分块阵
UB = -triu(A-DB);      %求A的上三角分块阵

N = 0;
tol = 1;
while tol>=eps 
    invDL = inv(DB-LB);
    x = invDL*UB*x0+invDL*b;    %块迭代公式
    N = N+1;
    tol = norm(x-x0);
    x0 = x;
    if(N>=M)
        disp('Warning: 迭代次数太多，可能不收敛！');
        return;
    end
end