题目:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
刷题链接在此: 下一个排列(力扣31题).
题目解析
参考这个博主的文章
链接: 一篇微信公众号文章.
讲得还算清晰,不过是用Python写的代码。但是可以参考思路~
此外参考力扣题解
主要思路是:交换a[i-1] 与 a[j] ,之后将a[i] 到最后的数翻转。得到的就是下一个排列。具体证明略。
代码:
1,普通解法
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if(nums.size()==1) return;
int i=nums.size()-1;
int j=nums.size()-1;
int pos;
while(i>0&&nums[i]<=nums[i-1])
{
i--;
}
pos=i-1;
if(pos>=0)
{
while(nums[j]<=nums[pos])
{
j--;
}
swap(nums[pos],nums[j]);
}
sort(nums.begin()+i,nums.end());
}
};
2,参考一篇题解,另一种想法
先进行了翻转
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int pos=nums.size()-1;
while(pos>0&&nums[pos]<=nums[pos-1])
{
pos--;
}
reverse(nums.begin()+pos,nums.end());
if(pos>0){
int i=pos;
for(;i<nums.size();i++)
{
if(nums[i]>nums[pos-1])
{
swap(nums[i],nums[pos-1]);
break;
}
}
}
}
};
3,第二种解法的优化版。引入了二分法寻找要交换的那个数
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int pos=nums.size()-1;
while(pos>0&&nums[pos]<=nums[pos-1])
{
pos--;
}
reverse(nums.begin()+pos,nums.end());
if(pos>0){
//利用二分法寻找pos到最后比num[pos-1]大的最小的那个数
auto i=upper_bound(nums.begin()+pos,nums.end(),nums[pos-1]);
swap(*i,nums[pos-1]);
}
}
};