matlab 验证魔方矩阵

function mofang(A)
[n,m]=size(A);
if n<3||m<3,disp('error');
elseif n~=m,disp('the size of M X must be N-by-N M');
else B=[sum(A,1),sum(A,2)',sum(diag(A))',sum(diag(rot90(A)))'];
    if B(1)==B(2)&B(2)==B(3)&B(3)==B(4)&B(4)==B(1),
        disp([B(1),B(2),B(3),B(4)]),
        disp('yes');
    else
        disp('no');
    end
end 

以上部分是声明定义函数部分，我们在命令行窗口可以输入:

>> A=[16,2,3,13;5,11,10,8;9,7,6,12;4,14,15,1];
>> check_mf(A)

输出产生的结果，每一行每一列主副对角线的总和

  34    34    34    34

  yes