31 next-permutation
题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
思路
找下一个排列,就是找到比当前排列大的最小值,那只要当前排列不是最大的,则会重排后面的某一部分(如果当前排列是最大值,那下一个排列就是最小值,是全部重排)。
我们举个例子,一个8位数(12345678的某个排列),后四位是这四个数能组成的最大值,前四位无所谓。比如68427531,那去调整后四位是没有意义的,只能调整第五位2,调整为后四位中比2大的最小值3,剩下四个数为7521,则组成最小值1257排在3后面,即68431257。
这样就很清晰了,从后往前找到第一个p[i-1] < p[i],将p[i-1]与它后面比大的最小值p[j]交换(由于p[i-1]的后面一定是降序的,所以从后往前找到第一个比它大的值就是p[j]),因为p[j-1]>p[j]>p[i-1]>p[j-1],所以交换后,p[i-1]之后的数仍然是降序的,这四位能组成的最小值是这四个数升序排列,那么反转这四位数就可以了。
class Solution {
public:
void nextPermutation(vector<int>& nums) {
//next_permutation(nums.begin(), nums.end());
int i=nums.size()-1;
while(i>0 && nums[i] <= nums[i-1])
{
i--;
}
if(i==0)
reverse(nums.begin(), nums.end());
else
{
int j = nums.size()-1;
while(nums[j] <= nums[i-1])
{
j--;
}
int tmp = nums[i-1];
nums[i-1] = nums[j];
nums[j] = tmp;
reverse(nums.begin()+i, nums.end());
}
}
};
思考
那前一个排列呢?仍然是一样的思考方式,前一个排列就是比当前值小的最大值,只要当前值不是最小的,就需要重排某一部分。比如31264578,找到第一个p[i-1] > p[i],与小于它的最大值交换,剩余部分逆序排列,6和5互换变成31254678,5后面的部分逆序,前一个排列即为31258764。