这是我以前写的一个程序,发布在一些Matlab的论坛,现在搬过来吧
clear;clc;close all
N=8;
c=ones(N);
% 行求和
blkele=num2cell(c,2);
A1=blkdiag(blkele{:});
% 列求和
A2=repmat(eye(N),1,N);
Aeq=[A1;A2];
beq=ones(2*N,1);
% 斜线情况判断
M=N-2;
A=zeros(4*M+2,N*N);
d{1}=arrayfun(@(i)find(diag(diag(c,i),i)),-M:M,'UniformOutput',0);
d{2}=arrayfun(@(i)find(fliplr(diag(diag(c,i),i))),-M:M,'UniformOutput',0);
d0=cat(1,d{:});
linearIndex=arrayfun(@(x)sub2ind(size(A),x+zeros(1,length(d0{x})),...
d0{x}'),1:numel(d0),'UniformOutput',0);
A(cell2mat(linearIndex))=1;
b=ones(4*M+2,1);
x=bintprog([],A,b,Aeq,beq);
Res=reshape(x,[],N);