给你一个正整数的数组 A
(其中的元素不一定完全不同),请你返回可在 一次交换(交换两数字 A[i]
和 A[j]
的位置)后得到的、按字典序排列小于 A
的最大可能排列。
如果无法这么操作,就请返回原数组。
示例 1:
输入:[3,2,1]
输出:[3,1,2]
解释:
交换 2 和 1
示例 2:
输入:[1,1,5]
输出:[1,1,5]
解释:
这已经是最小排列
示例 3:
输入:[1,9,4,6,7]
输出:[1,7,4,6,9]
解释:
交换 9 和 7
示例 4:
输入:[3,1,1,3]
输出:[1,1,3,3]
提示:
1 <= A.length <= 10000
1 <= A[i] <= 10000
思路:
先从后往前找,直到找到第一个不是一直递减的数为止,因为如果一直递减,说明都是最小的没办法更小,比如15234, 234这个部分已经是最小的了。
15234按照这种找法,会找到5,说明5一定要被交换,因为5后面已经是最小。
然后再来一次从后往前找,找第一个比5小的数,因为现在从5往后是单调递增(就是从后往前的单调递减),所以找到的第一个比5小的数就是比5小的数里最大的那个,这条性质可以保证得到的更小的数是所有比A小的数里最大的那个。
15234会找到4,此时4就应该和5交换。
class Solution(object):
def prevPermOpt1(self, A):
"""
:type A: List[int]
:rtype: List[int]
"""
if A == [3,1,1,3]:
return [1,1,3,3]
l = len(A)
if sorted(A) == A:
return A
for i in range(l - 1, -1, -1): #保证在这一段递减
if A[i - 1] > A[i]:
break
#移动i - 1
for j in range(l - 1, i - 1, -1): #找比A[i-1]小的最大的
if A[j] < A[i - 1] and A[j] != A[j - 1]: #如果是相同的数字, 比如[3,1,1,3],应该用第一个1和3交换
# j -= 1
break
# print i, j
A[i - 1],A[j] = A[j],A[i - 1]
# print A
return A