【题目】
给你一个正整数的数组 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,3,1,3]
解释:
交换 1 和 3
提示:
1 <= A.length <= 10000 1 <= A[i] <= 10000
【思路】
很明显,如果数组从小到大排序,则肯定是最小排列;否则,修改最后一个排序错误的数字。
具体来说,从后往前遍历数组,如果A[i] <= A[i+1],不用进行任何操作;否则,找到i后面最后一个小于A[i]的元素A[j](如有重复,则为第一个重复元素),最后替换A[i]和A[j]。
【代码】
python版本
class Solution(object):
def prevPermOpt1(self, A):
"""
:type A: List[int]
:rtype: List[int]
"""
if len(A) < 1:
return A
i = len(A) - 2
while i >= 0:
if A[i] > A[i+1]:
# 直接和最后一个元素比较
if A[i] > A[-1]:
A[i], A[-1] = A[-1], A[i]
else:
# 找到大于A[i]的第一个元素的前一个元素
j = i+2
while A[i] > A[j]:
j += 1
j -= 1
# 如果相邻的有重复元素,则必须找到重复元素第一个
while A[j] == A[j-1]:
j -= 1
A[i], A[j] = A[j], A[i]
break
i -= 1
return A