createm.m

来自「有兴趣学系matlab的朋友」· M 代码 · 共 24 行

function A=createM(x,y)
A=zeros((x+2)*(y+2),(x+2)*(y+2)+1);
for i=1:(x+2)*(y+2)
    A(i,(x+2)*(y+2)+1)=1;
end

k=[];
l=0;
while l<x
    k=[k,(x+4+l*(y+2)):(x+y+3+l*(y+2))];
    l=l+1;
end
i=1;
while i<=x*y
    A(k(i),k(i))=1;
    A(k(i),k(i)+1)=1;
    A(k(i),k(i)-1)=1;
    A(k(i)+y+2,k(i))=1;
    A(k(i)-y-2,k(i))=1;
    i=i+1;
end
l=[k,((x+2)*(y+2)+1)];
A=A(k,l);