在Heap的递归算法

时间:2015-07-07 08:19:37

标签: java algorithm recursion permutation

我已经实现了Heap算法来查找数组A元素的所有排列:

//A = {1, 2, 3, 4}; B = perms(A) ; num_row(B) = (4!+1) and B[0][0] = 4!;
//This is B.R. Heap's algorithm
public static void perms(int [] A, int [][]B, int n)
{
   if (n == 1)
   {
       int k = B[0][0];
       for (int i = 0; i < A.length; i++)
       {
           B[k + 1][i] = A[i];
       }
       B[0][0]++;
   }
   else
   {
       for (int i = 0; i < n - 1 ;i++)
       {
           perms(A, B, n-1);
           if (n % 2 == 0)
           {
               swap(A, i, n - 1);
           }
           else
           {
               swap(A, 0, n - 1);
           }
       }
       perms(A, B, n - 1); 
   }
}
public static void swap(int[] A, int i, int j)
{
    int temp = A[i];
    A[i] = A[j];
    A[j] = temp;
}

我是Java的新手。问题是我想把B作为函数perms(A)的输出(返回),但是在这个实现中,我必须初始化一个int [n! + 1] [A.length]调用函数前的B数组。我该怎么办? 是否有类似私有变量或java中的任何东西来帮助递归函数来记住前一个调用的变量?

由于

1 个答案:

答案 0 :(得分:1)

您可以创建一个“输入”方法来递归,如下所示:

public static int[][] perms(int[] a){
    int[][] perms = new int[factorial(a.length)+1][a.length];
    perms(a,perms,a.length);
    return perms;
}

方法factorial是众所周知的方法,例如可以在Google上找到 想知道n参数是否必要

修改

这不是必要的(以上更正)

修改

通过我的测试,k变量只是递增,所以我会使用这样的静态变量:

private static int counter = 0;
// your code here, following is a part of your perms method
if (n == 1)
{
    for (int i = 0; i < A.length; i++)
    {
        B[counter][i] = A[i];
    }
    counter++;
}
//and my code corrected too:
public static int[][] perms(int[] a){
    int[][] perms = new int[factorial(a.length)][a.length]; //+1 is not necessary
    counter=0; //necessary to call it again
    perms(a,perms,a.length);
    return perms;
}