Python对列表的一部分进行排序

Question

当我使用 sort 函数对列表末尾进行排序时，此用于查找整数列表的下一个排列的代码没有给出正确的答案。但是如果我使用 sorted 函数，它会给我正确的答案。为什么会这样？请有人帮我解决这个问题。

def nextPermutation(self, A):
    n = len(A)
    if n == 1:
        return A
    i = n - 2
    m = A[n - 1]
    while i >= 0:
        if A[i] < m:
            j = i + 1
            while j < len(A) and A[i] < A[j]:
                j += 1
            A[i], A[j - 1] = A[j - 1], A[i]
            A[i + 1 :].sort()  #Here if I use sorted it gives the correct answer
            return A
        else:
            m = max(A[i], m)
        i -= 1
    A.sort()
    return A

Answer 1

切片会创建列表的副本。当您说A[i + 1:].sort() 时，您正在创建A 的一些内容的副本，然后对它们进行排序。它不会影响A。试试这个：

B = A[i + 1:]
B.sort()
return B

当然，使用sorted() 和return sorted(A[i + 1:]) 会更容易，但从你的问题来看，我认为出于某种原因你对此不感兴趣。

Answer 2

好吧，在A[i+1:].sort() 行中，A[i+1:] 部分创建了一个新 列表对象。然后，.sort() 部分对该列表进行排序。然后，这个 new 对象被丢弃而不被分配给任何东西。因此，该行对您的代码没有影响。

我不确定你为什么会遇到所有这些麻烦，但是将类写成这样会更容易：

from itertools import permutations

class PermutationList:
    def __init__(self, pool):
        self.pool = pool
        self._generator = permutations(self.pool)

    def __iter__(self):
        while True:
            try:
                yield list(next(self._generator))
            except StopIteration:
                self._generator = permutations(self.pool)
                raise StopIteration

test_list = PermutationList([1,2,3,4])

for perm in test_list:
    print perm

# Output
# [1, 2, 3, 4]
# [1, 2, 4, 3]
# [1, 3, 2, 4]
# ...
# [4, 3, 2, 1]