题目描述
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
从数组末尾开始往前找,直到找到一个数a[i],满足a[i]<a[i+1],这个说明a[i]右侧的序列是从大到小排列的,也就是说以a[i]为首元素的排列这时是最大的,所以下一个排列就是包括a[i]在内的右侧元素的下一个排列。从a[i]右侧从尾部开始找到a[j],满足a[j]>a[i],交换a[i],a[j],此时a[j]右侧仍然满足逆序,所以需要将a[j]右侧所有元素颠倒顺序变为顺序,这样才是以a[j]为首元素的最小的排列,也是原数组的下一个排列。
过程动画
代码(c++)
class Solution {
public:
void reverse(vector<int>& nums,int i,int j){
while(i<j){
swap(nums[i],nums[j]);
i++;
j--;
}
}
void nextPermutation(vector<int>& nums) {
int i=nums.size()-2;
while(i>=0&&nums[i]>=nums[i+1]){
i--;
}
if(i>=0){
int j=nums.size()-1;
while(nums[j]<=nums[i]){
j--;
}
swap(nums[i],nums[j]);
}
reverse(nums,i+1,nums.size()-1);
}
};