function partitionsk(n,k)
%PARTITIONSK 产生n的所有的k分拆,个数即为整数n的k分拆数p_k(n)
% partitionsk( n, k ) 输入正整数n和k。程序依次输出所有按行降序排列的k分拆。
%基本思路:把分拆降序排列,首先增大前面的数(当然第1个数要相应地减少),一直增大到
%不能再增加为止,然后增加后面的数,同时把其前面的数(除第1个数之外)重置为与其相同,
%接着继续增加前面的数,......,一直做到连最后一个数都不能增大为止。
%$Author: WBC$ $Date: 2005/10/15$
if n<k
error(['第1个输入参数值 ' int2str(n) ' < 第2个输入参数值 ' int2str(k)]);
elseif k==1
disp(n);
else
%初始化
p=ones(1,k); %用于存储当前分拆的行向量,p除了第一个数外,其余数全为1
p(1)=n-k+1; %初始化p的第一个数
u=zeros(1,k); %当前分拆对应位的最大取值,比如u(i)表示分拆中第i个数的最大取值,
%显然它会随其后的p值的变化而变化,但与其前的p值的变化无关。u值在
%判断是否还可以继续增大时是必须的,每次都计算比较浪费机时,考虑到
%同一个u(i)值会多次使用,与其每次使用时都计算,不如只计算一次存储
%起来供多次使用(空间换时间)。
for i=2:k
u(i)=floor((n-k+i)/i);
end
disp(p); %输出当前分拆
if k==2
for i=p(2):u(2)-1
p(1)=p(1)-1;
p(2)=p(2)+1;
disp(p);
end
else
%它总是用于存储从第1个数开始到当前位置的所有p值之和,作用是产生新的u
s=n-k+2; %这里s=p(1)+p(2)
while 1
if p(2)<u(2) %判断是否可以通过增大p(2)而获得不同分拆
for i=p(2):u(2)-1 %通过增大p(2)减少p(1)而获得不同分拆,并输出之
p(1)=p(1)-1;
p(2)=p(2)+1;
disp(p);
end
end
%找一个位置l:p(l)未达上界
l=3;
while p(l)>=u(l)
s=s+p(l);
l=l+1;
if l>k %l>k表明没有适当的位置,其p值可以增加了,即已经找到了所有分拆
return; %程序终止
end
end
p(l)=p(l)+1; %给未达上界的p(l)增加1
s=s-1; % l位置之前的p值之和减1,因为p(l)增加了1
%由于p(l)发生变化,所以其前的u可能发生变化,该循环就是来重置u。
%同时也重置p。
for i=l-1:-1:2
p(i)=p(l); %重置p
u(i)=floor(s/i); %重置u
s=s-p(l);
end
p(1)=s;
s=s+p(2); %在下一次循环开始时保证s=p(1)+p(2)
disp(p);%输出重置后的新p
end
end
end