Matlab 枚举法求解八皇后问题

clear;clc;close all
cols=perms(1:8);  % 皇后所在列的位置
S=size(cols,1);
M=zeros(8,8,S); % 存储所以情况的矩阵
M(sub2ind(size(M), repmat((1:8)',1,S), cols', repmat(1:S,8,1))) = 1;
M(:,:,arrayfun(@(k)any([sum(spdiags(M(:,:,k))),sum(spdiags(rot90(M(:,:,k))))]>1),1:S)) = [];