LeetCode 31. 下一个排列
题目描述
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
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
思路解析
该题的意思可能比较难理解。意思是找到一个排列,他组成的数字刚好大于给定的排列。
- 从数组末尾开始遍历,如果数组从尾至头一直是递增的,则该排列为降序排列,直接输出该数组的升序排列即可(
std::sort
) - 若该排列从第\(i\)位开始至结尾是递减的,则我们需要将第\(i - 1\)位与其后所有数字中,刚好大于第\(i - 1\)位的数字进行交换。
- 将第\(i\)位至末尾按升序排序。
代码实现
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int i;
for(i = nums.size() - 1; i > 0; i--) {
if(nums[i] > nums[i - 1])
break;
}
if(i == 0) {
sort(nums.begin(), nums.end());
return;
}
int target = nums[i - 1];
// curVal 记录 nums 第 i 位以后的刚好大于 nums[i - 1]的值
// curInd 记录 curVal 对应的下标
int curVal = nums[i]; // nums[i] > nums[i - 1]
int curInd = i;
for(int j = i; j < nums.size(); j++) {
if(nums[j] > target && nums[j] < curVal) {
curInd = j;
curVal = nums[j];
}
}
// 交换 nums[i - 1] 与 其后刚好大于 nums[i - 1]的数字
nums[curInd] = nums[i - 1];
nums[i - 1] = curVal;
// 对 nums[i] 以后的数字排序
sort(nums.begin() + i, nums.end());
return;
}
};