题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
题目的意思是,找出排列数的下一个比其大的数,若其为最大则其下一个为最小。列如,将[1,2,3]三个数全排列有以下123,132,213,231,312,321,若当前为231,则下一个为312,若当前为321,则下一个为123,以此类推。
思路
对于这中题目,我们可以先设定一组数据进行测试,找出其中的规律。
假设一组数据为1376542,则其下一个为1423567,怎么得来的呢。
可以看到数字从后往前看都是依次增大,直到3位置才减少,然后在从后面往前找到一个比3大的数,交换这两个数,最后将3后面的数倒置,就能得到下一个的值。
1 3 7 6 5 4 2
1 4 7 6 5 3 2
1 4 2 3 5 6 7
可以按照上述思路编写代码,数组问题考虑使用双指针,一个定义为最后一个的下标,记为j(用来第二次从后往前找),最后一个定义为倒数第二个的下标,记为i(用来第一次找到减少的值)。第一次从后往前迭代找到第一个减少的数,即nums[i]。然后在从后面往前迭代找到比第一个减少的位置的数大的数,即nums[j],交换这两个数,然后将i后面的值都进行倒置,即为结果。
代码
class Solution {
public:
void nextPermutation(vector<int>& nums)
{
if(nums.size()<=1)//为空或只有一个值,直接返回
return;
//i用来从后往前找到第一个递减的值
//j用来从后往前找到第一个比nums[i]大的值
int i=nums.size()-2,j=nums.size()-1;
while(i>=0 && nums[i] >= nums[i+1])//迭代找到第一个递减的值
--i;
if(i>=0)//若i为0,则当前值为最大,返回最小的值
{
while(j>=0 && nums[j]<=nums[i])//迭代找到第一个比nums[i]大的数
--j;
swap(nums[i],nums[j]);//找到后交换
}
for (int k = i + 1; k < (nums.size()+i+1) / 2; k++)//倒置,及找到中心点将两边交换,亦可使用下面标准库函数
swap(nums[k], nums[nums.size()+i - k]);
//reverse(nums.begin()+i+1,nums.end());
}
};