题目描述
给你一个正整数数组 arr(可能存在重复的元素),请你返回可在 一次交换(交换两数字 arr[i] 和 arr[j] 的位置)后得到的、按字典序排列小于 arr 的最大排列。
如果无法这么操作,就请返回原数组。
示例 1:
输入:arr = [3,2,1]
输出:[3,1,2]
解释:交换 2 和 1
示例 2:
输入:arr = [1,1,5]
输出:[1,1,5]
解释:已经是最小排列
示例 3:
输入:arr = [1,9,4,6,7]
输出:[1,7,4,6,9]
解释:交换 9 和 7
提示:
1 <= arr.length <= 104
1 <= arr[i] <= 104
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/previous-permutation-with-one-swap
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
本题只允许交换两个元素且交换一次。得到按字典序排列小于 arr 的最大排列。
只需要从后往前遍历,找到不符合降序排列的第一个元素arr[i],此时arr[i+1]~arr[length-1]都是按照升序排列,该arr[i]就是需要被交换的元素,另一个元素从arr[i+1]-arr[length-1]中找,是大于arr[i]的最小元素,如果相等就选取下标在更前面的那个。
代码
class Solution {
public int[] prevPermOpt1(int[] arr) {
int l=arr.length;
//找第一个被交换元素
for(int i=l-1;i>0;i--){
if(arr[i]>=arr[i-1]) {
continue;
}
int dex=i-1;//第一个被交换元素下标
//如果被交换元素比最后一个元素还大,直接交换该元素与最后一个元素,如{5,2,3,4}这种情况
if(arr[l-1]<arr[dex]){
int temp=arr[dex];
arr[dex]=arr[l-1];
arr[l-1]=temp;
return arr;
}
//找第二个元素
for(int j=i;j<l;j++){
if(arr[j]<arr[dex] && arr[j+1]>=arr[dex]){//找比第一个元素小的最大的那个,如{5,3,4,6},dex2指向元素4
int dex2=j;
//如果相等,第二个元素是下标在前面的那个,如{5,4,4,6},dex2指向第一个4
for(int k=dex2;k>=i;k--){
if(arr[dex2]==arr[k]){
dex2=k;
}
}
int temp=arr[dex];
arr[dex]=arr[dex2];
arr[dex2]=temp;
return arr;
}
}
}
return arr;
}
}
看了官方题解,发现我写的是一坨,找第二个元素指只需下面一行代码即可
/**
*只要比arr[i]大(arr[i]是第一个被交换元素),
*或者和他下标在前面的元素相等,直接跳过,执行j--,最终定位到第二个元素下标j
*/
while (arr[j] >= arr[i] || arr[j] == arr[j - 1]) {
j--;
}