Question

作为Generate a sequence of all permutation of some range of numbers的后续内容，我在Perm类中编写了以下代码：

/**
 * Permute A to its next permutation, if possible. Returns true if there is
 * such a permutation, and false otherwise.
 */
static boolean nextPerm(int[] A) {
    int N = A.length;
    int k = N - 1;
    int v;
    Set<Integer> S = new HashSet<Integer>();

    while (k >= 0) {
        int max = Collections.max(S);
        if (max > A[k]) {
            v = Collections.min(S);
            S.remove(v);
            S.add(A[k]);
            A[k] = v;
            int [] sArr = convertToArray(S);
            for (int i = k + 1; i < N - 1; i += 1) {
                A[i] = sArr[i - k - 1];
            }
            return true;      
        } else {
            S.add(A[k]);
            k -= 1;
        }
    }
    return false;
}

static int [] convertToArray (Set<Integer> s) {
    int [] sArr = new int[s.size()];
    int index = 0;
    for(Integer i : s) {
        sArr[index++] = i;
    }
    Arrays.sort(sArr);
    return sArr;
}

基本上，它的作用是生成一系列数字的所有排列序列，如下所示：

Let A be a sequence of integers 0 to N-1 in ascending order 
(let's assume its an array of int[N]).    

next_permutation(A):
    k = N-1
    S = { }
    while k >= 0:
        if S contains a value larger than A[k]:
            v = the smallest member of S that is larger than A[k]
            remove v from S
            insert A[k] in S
            A[k] = v
            A[k+1:N-1] = the values in S in ascending order.
            return true
        else:
            insert A[k] in S
            k -= 1
    return false

我的代码似乎不起作用。有人可以点灯吗？谢谢！

更新：在接受了每个人的意见并稍微解决了问题后，我能够让它发挥作用！我学到了一些东西：

正如Worakam所提到的，TreeSet（与HashSet相比）在这个问题中非常方便，因为它已经排序并具有更高的（）函数。
最初我认为将TreeSet转换为Integer数组会很忙，因为Integer对象不是int。然而，事实证明（可能是由于java5之后的自动装箱/拆箱），我能够将Integer数组中的元素视为普通的int并将int项添加到它（如for循环中所示）。

以下是工作代码：

static boolean nextPerm(int[] A) {
    int N = A.length;
    int k = N - 1;
    int v;
    int max = 0;

    TreeSet<Integer> S = new TreeSet<Integer>();

    while (k >= 0) {
        if (!S.isEmpty() && S.last() > A[k]) {
            v = S.higher(A[k]);
            S.remove(v);
            S.add(A[k]);
            A[k] = v;
            Integer [] sArr = new Integer[S.size()];
            S.toArray(sArr);

            for (int i = k + 1; i < N; i += 1) {
                A[i] = sArr[i - k - 1];
            }
            return true;
        } else {
            S.add(A[k]);
            k -= 1;
        }
    }
    return false;
}

非常感谢大家!!

Answer 1

首先，Collections.max(S)在集合为空时抛出NoSuchElementException。

其次，选择S的最小成员并不是实现“S的最小成员，大于A [k]”的正确方法。

我建议您不要使用HashSet，而应使用排序的数据结构，例如java.util.TreeSet。它将消除自己对集合进行排序的需要。方法higher()对您的需求非常有用。

生成一系列数字的所有排列的序列第二部分

1 个答案: