Matlab 函数求解线性方程组 Ax=b

function x = linear_solve(a,b) % a: n x n matrix % b: n x 1 vector % x: n x 1 solution vector

% check if a is square matrix if size(a,1)~=size(a,2) error('a must be a square matrix'); end

% check if b is a column vector if size(b,2)~=1 error('b must be a column vector'); end

% check if the dimensions of a and b are compatible if size(a,1)~=size(b,1) error('dimensions of a and b are not compatible'); end

% solve the linear system using the backslash operator x = a\b;

end