leetcode31 下一个排列(Medium)
题目来源:leetcode31 下一个排列
题目描述:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解题思路:
找到a[i-1]<a[i]的最大下标,即最后一个上升的位置pos=i-1。然后从右到左边找到第一个比它大的位置,两者交换。再把pos后面的字母升序排序。
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int pos=-1;
//找到最后一个上升的位置
for(int i=nums.size()-1;i>=1;i--){
if(nums[i-1]<nums[i]){
pos=i-1;
break;
}
}
//没找到,说明现在已经是完全逆序了,改成升序。
if(pos==-1) {
sort(nums.begin(),nums.end());
return;
}
//存在,则找到pos之后,最后一个比他大的位置,然后交换两者
for(int i=nums.size()-1;i>pos;i--){
if(nums[i]>nums[pos]) {
swap(nums[i],nums[pos]);
break;
}
}
//然后把pos后的值升序排序
sort(nums.begin()+pos+1,nums.end());
}
};