matlab生成子集,生成集合[n]的所有k-子集MATLAB代码

function ss=ksubsets(n,k)

%KSUBSETS   生成集合[n]的所有k-子集

%    ss = ksubsets( n , k )    把集合[n]的所有k-子集按行存入ss返回

%$Author: WBC$     $Date: 2005/9/30$     $ref: 卢开澄 卢华明  组合数学(第3版)  p30-31$

%----------两种特殊情况-----------

if k==0,  ss=[];   return;   end

if k==n,  ss=1:n;   return;   end

%----------其余情况-------------

cnk=prod(n-k+1:n)/prod(1:k);

ss=zeros(cnk,k);

b=n-k+(1:k);

s=1:k;

loop=1;

ss(loop,:)=s;

j=k; %当前活动的位置, 也就是说可以把该位置的数未达到上界

%主循环

while j>0

while s(j)>=b(j)    % 该循环保证j是活动的

j=j-1;

end

s(j)=s(j)+1;%我们必须把j对应元素加1, 从而得到下一个k-子集

%对于j不在最末的情形

if jif s(j)s(j:end)=s(j):(s(j)+k-j);

j=k;

else   % 如果已经加1的位置j处的元素刚好达到上界

s(j:end)=b(j):n;

j=j-1;

end

end

loop=loop+1;

ss(loop,:)=s;

end