31. 下一个排列
时间:2020年11月10日
知识点:next_permutation实现、数组
题目链接:https://leetcode-cn.com/problems/next-permutation/
题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解法:
- C++的标准库函数 next_permutation 中就是用这样的方法
- 首先从后向前找第一个顺序对(i,i+1),满足 a[i] < a[i+1]。这样「较小数」即为 a[i]。此时 [i+1,n)必然是下降序列。
- 如果找到了顺序对,那么在区间 [i+1,n) 中从后向前查找第一个元素 j 满足 a[i] < a[j]。这样「较大数」即为 a[j]。
- 交换 a[i]与 a[j],此时可区间[i+1,n) 必为降序。我们可以直接使用反转区间 [i+1,n) 使其变为升序,而无需对该区间进行排序。
案例:
3 2 4 1
a[i] = 2
a[j] = 4
交换a[i]和a[j] 3 4 2 1
反转区间 3 4 1 2
代码
#include <stdio.h>
#include <vector>
#include <iostream>
#include <stack>
using namespace std;
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int idx = nums.size()-2;
while(idx >=0 && nums[idx] >= nums[idx+1])
idx--;
if(idx >=0 ){
int idy = nums.size()-1;
while(idy >=0 && nums[idy] <= nums[idx])
idy--;
swap(nums[idx], nums[idy]);
}
reverse(nums.begin()+idx+1, nums.end());
}
};
int main()
{
vector<int> nums{1,1};
Solution s;
s.nextPermutation(nums);
for(int x:nums)
cout<<x<<endl;
return 0;
}
今天也是爱zz的一天哦!