生成x个整数分区

时间:2014-07-20 14:54:54

标签: java algorithm math integer combinatorics

我有一个递归算法,它生成一个作为参数给出的数字的所有组合。它也可以基于'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);
            }
        }
    }
}

1 个答案:

答案 0 :(得分:0)

引入全局计数器(与h位于同一位置),将其初始化为零。每次向答案添加分区时,将计数器增加1.在递归调用GenP后,检查计数器是否已达到x,如果已经,则立即从递归函数返回。

您的norep版本不容易修补。你的算法实际上是否会发出重复数据? (您发布的内容不是完整且可运行的Java代码,因此我没有运行它。)如果确实如此,那么可以修改算法本身以仅发出唯一的分区。但是没有明确说明具体的限制因素(或者至少我不能一目了然)。一旦指定了问题的确切公式,一个不产生重复的干净而有效的算法可能是一个单独问题的主题。