代码总结在github仓库:
https://github.com/Rayholle/LeetCode-Newcoder/blob/master/STL/下一个更大的排列.cpp
…今天做了一道挺有意思的题目,leetcode原题描述如下:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。 如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。 必须原地修改,只允许使用额外常数空间。 以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
所以该怎么办呢?
按正常思路当然能想到下一个更大的排列就要从后向前遍历数组,比如1,2,3->1,3,2,就是从尾部开始,与其前一个位置数字比较,弱小于前一个数值,交换即可得到比原排列更大的下一种排列1,3,2。
但是考虑更复杂一点的情况:
1,2,3,6,5,4 -> 1,2,4,3,5,6 (1)
1,2,3,6,5,2 -> 1,2,5,2,3,6 (2)
对于第一种情况,尾位’4’不大于’5’,继续前向遍历;‘5’不大于’6’,继续前向遍历;‘6’大于’3’,此时可以将尾位’4’与’3’交换,并将数组从’6’位置开始向后排序,组成最小值序列即可,就得到最终的1,2,4,3,5,6。
可是同样的方法应用在(2)中却不可行,因为在交换尾位’2’和’3’时,数组排列的值变小了(2<3),所以在交换时需要条件判断,尾位值与被交换值的大小。(1)中为>情况,(2)中位<情况,此时需要找出从’6’下标到尾位之前的数中最小的且大于’6’之前的’3’的值,并将其与’3’交换即可。
推广到一般情况:从数组倒着查找,找到nums[j] 比nums[j-1]大的时候,就将nums[j-1]与j(下标)到nums.size()-1(下标)(闭区间)当中找到一个最小的且比nums[j-1]大的元素交换。交换后,再把nums[j]到nums[nums.size()-1]排序
下面贴出我的代码:
void nextPermutation(vector<int>& nums){
int length = nums.size();
if(length == 0 || length == 1)
return;
int j = length -1;
for(; j>0; j--){
if(nums[j] > nums[j-1]){//若找到可交换点,寻找后续的另一个交换点
int min = j;
for(int k = j+1; k < length; k++){
if(nums[min]>nums[k] && nums[j-1]<nums[k]) min = k;//j后续下标中最小且大于j-1的下标
}
//交换min与j-1的value
int temp = nums[j-1];
nums[j-1] = nums[min];
nums[min] = temp;
break;//交换完 跳出循环
}
}
//从j位置开始插入排序
for(int i = j+1; i <length; i++){
int temp = nums[i];
int k = i-1;
for(; k >=j&&nums[k]>temp; k--) nums[k+1] = nums[k];
nums[k+1] = temp;
}
}
希望广大网友能够提供更好的思路和代码啦