leetcode 31. 下一个排列 medium
题目描述:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解题思路:
要求是下一个更大的排列,所以肯定是对这个数组越靠右部分调整,那越跟现在这个排列接近。
然后我们观察到对于任何给定序列的降序,是不可能有更大的排列。
例如,以下数组不可能有下一个排列:
[9, 5, 4, 3, 1]
所以,我们从后面往前遍历,找到第一个满足a[i]<a[i+1]的i,现在对包括a【i+1】在内的右侧子数组重新排列是不可能创建更大的排列,因为该子数组由数字按降序组成。因此,我们需要重新排列包括a[i]在内的子数组。
那么,什么样子的重新排列将产生下一个更大的数字呢?我们想要创建比当前更大的排列(且是最小的)。因此,我们需要将a[i] 跟 后面最后一个比它大的数swap(也就是从后面往前遍历的第一个),然后reverse一下a[i+1]到最后形成的子数组。
总结:
- 从右往左找到第一个非逆序的,即使得 a
[i] < a[i+1]
成立的 i- 从右往左找到第一个大于a【i】的数 a【j】,swap(a【i】,a【j】)
- 将a【i】后面的所有数逆序,变为以小到大的顺序。
代码:
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if(nums.size()<2)
return;
int len=nums.size();
for(int i=len-2;i>=0;--i){
if(nums[i]<nums[i+1]){
int j=len-1;
for(;nums[j]<=nums[i];--j) ;
swap(nums[i],nums[j]);
reverse(nums.begin()+i+1,nums.end());
return;
}
}
reverse(nums.begin(),nums.end()); // 找不到,则按题目要求 reverse
return;
}
};