我想生成所有可能的增加的数字子序列(允许重复),从1到n,但长度为k。
防爆。 n = 3,k = 2
输出:
1 1
1 2
1 3
2 2
2 3
3 3
这是我的代码:
#include <stdio.h>
int s[100];
int n=6;
int k=4;
void subk(int prev,int index)
{
int i;
if (index==k)
{
for(int i=0; i<k; i++)
printf("%d ",s[i]);
printf("\n");
return;
}
s[index]=prev;
for (i=prev; i<=n; ++i)
{
subk(i,index+1);//,s,n,k);
}
}
int main()
{
int j;
for (j = 1; j<=n ; ++j)
{
subk(j,0);
}
return 0;
}
但这会产生一些不必要的重复。我如何消除这些?
答案 0 :(得分:2)
我已使用n = 3
和k = 2
测试了您的代码,并获得了以下结果:
1 1
1 1
1 1
1 2
1 2
1 3
2 2
2 2
2 3
3 3
这显然是不正确的,因为有几个相同的数字,如1 1
或1 2
。
但到底出了什么问题?
如果n = 3
和k = 3
,请写下正确的结果。现在将这些与我们在n = 3
和k = 2
时获得的结果进行比较。
correct program (incorrect)
k = 3 k = 2
1 1 1 1 1
1 1 2 1 1
1 1 3 1 1
1 2 2 1 2
1 2 3 1 2
1 3 3 1 3
2 2 2 2 2
2 2 3 2 2
2 3 3 2 3
3 3 3 3 3
现在我们可以看到,当我们设置k = 3
时,程序的错误输出与正确答案的前两列相同。这意味着如果我们设置k = 2
,程序将解决3列的问题,但只显示前两列。
您需要停止程序多次写入相同的数字。
执行此操作的一种方法是在将最后一个数字(index == (k - 1)
)写入缓冲区s
时,仅在subk-function中执行for-loop一次。
为了实现这一点,您需要在for循环的末尾添加以下两行。
if (index == (k - 1))
break;
(而不是 break 你也可以使用 return )
添加这两行后,该函数应如下所示:
void subk(int prev, int index)
{
int i;
if (index == k)
{
for (int i = 0; i<k; i++)
printf("%d ", s[i]);
printf("\n");
return;
}
s[index] = prev;
for (i = prev; i <= n; ++i)
{
subk(i, index + 1);//,s,n,k);
if (index + 1 == k)
break;
}
}
解决问题的另一种方法是将行s[index] = prev;
移到函数的开头,并将if语句中的k
更改为k - 1
。
现在该函数应如下所示:
void subk(int prev, int index)
{
int i;
s[index] = prev;
if (index == k - 1)
{
for (int i = 0; i<k; i++)
printf("%d ", s[i]);
printf("\n");
return;
}
for (i = prev; i <= n; ++i)
{
subk(i, index + 1);//,s,n,k);
}
}
使用此解决方案,当index
显示程序位于最后一个“子编号”时,永远不会执行for循环。它只显示数字并退出函数,因为返回。
两种解决方案都能得到正确的结果,但我个人更喜欢第二种解决方案,因为在for循环的每次迭代中都没有执行额外的if语句,而且程序(稍微)更快。