【leetcode】969. Pancake Sorting

题目如下：

Given an array A, we can perform a pancake flip: We choose some positive integer k <= A.length, then reverse the order of the first k elements of A. We want to perform zero or more pancake flips (doing them one after another in succession) to sort the array A.

Return the k-values corresponding to a sequence of pancake flips that sort A. Any valid answer that sorts the array within 10 * A.length flips will be judged as correct.

Example 1:
Input: [3,2,4,1]
Output: [4,2,4,3]
Explanation: 
We perform 4 pancake flips, with k values 4, 2, 4, and 3.
Starting state: A = [3, 2, 4, 1]
After 1st flip (k=4): A = [1, 4, 2, 3]
After 2nd flip (k=2): A = [4, 1, 2, 3]
After 3rd flip (k=4): A = [3, 2, 1, 4]
After 4th flip (k=3): A = [1, 2, 3, 4], which is sorted. 
Example 2:
Input: [1,2,3]
Output: []
Explanation: The input is already sorted, so there is no need to flip anything.
Note that other answers, such as [3, 3], would also be accepted.
Note:

1 <= A.length <= 100

A[i] is a permutation of [1, 2, ..., A.length]

解题思路：本题没有要求求出最少的操作次数，而且规定了操作上限是 10 * A.length。我的方法是首先把最大的数移到最后，这里只需要两步，第一是找出最大的数的位置，并且把这一段反转，这样的话最大数就在第一位了，接下来逆转整个数组，最大数就被翻转到最后。接下来是次大数，方法也一样，只不过在操作的过程中不处理数组最后一个元素（即已经放好位置的最大数）。这种方法理论上的最大操作次数是 2 * A.length。

代码如下：

class Solution(object):
    def pancakeSort(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        bound = len(A)
        sa = sorted(A)
        end = len(A)
        res = []
        while A != sa:
            inx = A.index(bound)
            if inx != 0:
                A = A[0:inx+1][::-1] + A[inx+1:]
                res.append(inx+1)
            A = A[:end][::-1] + A[end:]
            res.append(end)
            end -= 1
            bound -= 1
        return res

【leetcode】969. Pancake Sorting

猜你喜欢