我有一个递归算法,它生成一个作为参数给出的数字的所有组合。它也可以基于'k'进行分区,也可以作为参数给出。只要我们输入较小的数字,它就可以正常工作。但随着'n'的增加,计算结果需要更多的时间和空间。
是否可以给出'x'作为输入,这样算法只返回数字的x分区,而不是全部。这是我正在寻找的一个例子:
输入: n = 10, k = 4,将n分成'k'部分 x = 2,所需的分区数 m = 4,分区中的最大数量
输出:
1)4,2,2,2
2)4,3,2,1
以下是我使用的算法:
int h=0; //iterator
public ArrayList<int[]> generate_partitions(int n,int k,int max,boolean norep)
{
int korig;
korig = k;
int[] A = new int[korig+1];
ArrayList<int[]> partitions = new ArrayList<int[]>();
GenP(A, n, k, korig, 1,partitions,max);
if(norep)
{
for(int i=0; i<partitions.size(); i++)
{
if(check_repetition(partitions.get(i),max))
partitions.remove(i);
}
}
return partitions;
}
boolean check_repetition(int[] a,int max)
{
boolean[] hash = new boolean[max+1];
for(int i=0; i<max+1; i++)
hash[i]= false;
for(int i=0; i<a.length; i++)
{
if(hash[a[i]]==false)
hash[a[i]]=true;
else
return true;
}
return false;
}
void GenP(int[] A, int n, int k, int korig, int l, ArrayList<int[]> partitions,int max)
{
//n = number to partition
//korig = original k
//l = least number integer required in partition
if (k==1) // k = number of partitions
{
A[k]=n;
int [] temp = new int[korig];
// System.out.println("size = "+korig);
boolean max_check = false;
for (int j=1; j<=korig; j++)
{
// System.out.print(A[j]+" ");
temp[j-1]=A[j];
if(A[j]>max)
max_check = true;
}
if(!max_check) {
partitions.add(temp);
}
//System.out.println();
}
else
{
if (k==0)
{
h=0;
}
else
{
h=n/k;
for (int i=l; i<=h; i++)
{
A[k]=i;
GenP(A, n-A[k], k-1, korig, A[k], partitions,max);
}
}
}
}
答案 0 :(得分:0)
引入全局计数器(与h
位于同一位置),将其初始化为零。每次向答案添加分区时,将计数器增加1.在递归调用GenP后,检查计数器是否已达到x
,如果已经,则立即从递归函数返回。
您的norep
版本不容易修补。你的算法实际上是否会发出重复数据? (您发布的内容不是完整且可运行的Java代码,因此我没有运行它。)如果确实如此,那么可以修改算法本身以仅发出唯一的分区。但是没有明确说明具体的限制因素(或者至少我不能一目了然)。一旦指定了问题的确切公式,一个不产生重复的干净而有效的算法可能是一个单独问题的主题。