【Leetcode】969. Pancake Sorting 煎饼排序

解法

解法一：BIT

当前k个数已经有序时，假设前k个数里有i个比第k+1个数大，我们可以：
[1:k-i][k-i+1:k][k+1]翻转全部->
[k+1][k:k-i+1][k-i:1]翻转前i+1个数->
[k-i+1:k][k+1][k-i:1翻转前i个数->
[k:k-i+1][k+1][k-i:1]翻转全部->
[1:k-i][k+1][k-i+1:k]
可以用BIT统计有多少个数比第k+1个数小
好处是不需要真的翻转数组

class Solution(object):
    def pancakeSort(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        ans = []
        n = len(A)
        c = [0]*n
        def lowbit(k):
            return k&-k
        def add(k):
            while k<=n:
                c[k-1] += 1
                k += lowbit(k)
        def sum(k):
            res = 0
            while k>=1:
                res += c[k-1]
                k -= lowbit(k)
            return res
        for i,a in enumerate(A):
            le = sum(a)
            if le==0 and i!=0:
                ans += [i,i+1]
            elif le<i:
                ans += [i+1,i-le+1,i-le,i+1]
            add(a)
        return ans

解法二：每次归位最大值

参考评论区@powcai的答案

当n=k时，找到最大值应该在的位置，然后把最大值转到第一个，然后再整体反转，这样最大值就归位了
问题转化为n=k-1的时候的子问题

class Solution(object):
    def pancakeSort(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        ans = []
        n = len(A)
        while n:
            idx = A.index(n)
            if idx!=n-1:
                ans += [idx+1,n]
                A = A[idx+1:n][::-1]+A[:idx+1]+A[n:]
            n -= 1
        return ans