【LeetCode】下一个排列【找规律】

时间:2023-03-08 20:34:28

实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

必须原地修改,只允许使用额外常数空间。

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。

1,2,3 → 1,3,2

3,2,1 → 1,2,3

1,1,5 → 1,5,1

来源:力扣(LeetCode)

链接:https://leetcode-cn.com/problems/next-permutation

分析:

方法1:直接调API,next_permutation函数

ps:不建议采用这种做法

方法2:寻找规律

1.先找到最大的索引k,满足v[k]<v[k+1],如果不存在,则说明当前序列是54321这种序列,其下一个序列是12345,直接翻转整个数组

2.再找到另外一个最大的索引L,满足v[L]>v[k]

3.交换v[L]和v[k]

4.翻转v[k+1,end]

总结起来就是一句话:从后往前遍历找到第一个降序值x,然后从后往前找到第一个比该值大的值y,交换着两个值,最后把该值后面的值逆转即可

时间复杂度:O(N)

空间复杂度:O(1)

class Solution {
public:
void nextPermutation(vector<int>& v)
{
int n=v.size();
if(n==1)
return ;
int flag=0;
int k;
for(int i=n-2; i>=0; i--)
{
if(v[i]<v[i+1])
{
flag=1;
k=i;
cout<<k<<endl;
break;
}
}
if(flag==0)
{
reverse(v.begin(),v.end());
return ;
}
int l;
int x=v[k];
for(int i=n-1; i>=k; i--)
{
if(v[i]==x)
{
l=i;
x=v[i];
}else if(v[i]>x)
{
l=i;
break;
}
}
swap(v[k],v[l]);
reverse(v.begin()+k+1,v.end());
}
};