题目
思路
这道题的要求初看还是比较复杂的:既要字典序小于原来的序列,又要在小于的字典序中求出最大值。按照题目的要求,应该尽量交换排在后面的元素,因为交换越到后面的元素所产生的的字典序“增量”越小,也就更接近与题目的要求。而题目中需要交换的地方即出现了逆序的地方(即i<j且A[i]>A[j]),如果数组中有这样的逆序则证明可以通过交换元素使字典序减小,从而符合题目要求。但是最终的答案并不是直接交换相邻的元素,而是在发现一个相邻的逆序对后,记录当前较大的值并继续向后搜索,在这个遍历过程中找到最大的小于当前较大值的数,最后再进行交换。
在算法实现时我是使用了从前往后遍历的方法。因此在遇到新的相邻逆序对时需要更新较大值的记录并重新搜索,代码如下:
class Solution:
def prevPermOpt1(self, arr: List[int]) -> List[int]:
n=len(arr)
if n==1:
return arr
re1=-1
re2=-1
temp=arr[0]
ans=0
for i in range(n-1):
#print(re1,re2)
if arr[i]>arr[i+1]:
re1=i
re2=i+1
temp=arr[re2]
else:
if re1!=-1:
if temp<arr[i+1] and arr[i+1]<arr[re1]:
temp=arr[i+1]
re2=i+1
#print(re1,re2)
if re1==-1:
return arr
else:
arr[re1],arr[re2]=arr[re2],arr[re1]
return arr
更快的算法是从后往前遍历,这样找到的第一个逆序对就是在最后面的逆序对,不必像之前的方法那样反复尝试。