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

来源:互联网 发布:java 支付宝支付 demo 编辑:程序博客网 时间:2024/06/14 02:31

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 j<k
        if s(j)<b(j)   % 如果已经加1的位置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