题目:
给你一个整数数组 arr ,请使用 煎饼翻转 完成对数组的排序。
一次煎饼翻转的执行过程如下:
- 选择一个整数 k ,1 <= k <= arr.length
- 反转子数组 arr[0…k-1](下标从 0 开始)
例如,arr = [3,2,1,4] ,选择 k = 3 进行一次煎饼翻转,反转子数组 [3,2,1] ,得到 arr = [1,2,3,4] 。
以数组形式返回能使 arr 有序的煎饼翻转操作所对应的 k 值序列。任何将数组排序且翻转次数在 10 * arr.length 范围内的有效答案都将被判断为正确。
思路:类选择排序
设一个元素的下标是index,我们可以通过两次煎饼排序将它放到尾部:
- 选择 k=index+1,然后反转子数组arr[0…k−1],此时该元素已经被放到首部。
- 选择k=n,其中 n 是数组 arr 的长度,然后反转整个数组,此时该元素已经被放到尾部。
通过以上两步操作,我们可以将当前数组的最大值放到尾部,然后将去掉尾部元素的数组作为新的处理对象,重复以上操作,直到处理对象的长度等于一,此时原数组已经完成排序,且需要的总操作数是2×(n−1),符合题目要求。如果最大值已经在尾部,我们可以省略对应的操作。
解答:
class Solution:
def pancakeSort(self, arr: List[int]) -> List[int]:
n=len(arr)
res=[]
def reverse(start,end):
while start<end:
arr[start],arr[end]=arr[end],arr[start]
start+=1
end-=1
for i in range(n-1,-1,-1):
#当前位应放置的数为cur=i+1
cur=i+1
if arr[i]!=cur:
idx=arr.index(cur)
#将cur置于第一位
if idx!=0:
res.append(idx+1)
reverse(0,idx)
#将cur置于当前位
res.append(cur)
reverse(0,i)
return res