LeetCode_31_下一个排列
原创
©著作权归作者所有:来自51CTO博客作者晴天码字的原创作品,请联系作者获取转载授权,否则将追究法律责任
题目描述:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
算法思想:从后向前遍历数组,找到第一个不满足降序排列的元素,即k和k+1是升序,该元素就是需要调换位置的元素。
如果从后向前没有找到不满足降序的元素,则说明整个数组已经按照从前向后降序排列。将整个数组reverse即可
如果找到了k的位置,然后定义j,从后向前查找第一个大于k的下标j,swap(nums[k],nums[j]),这样从原来的下标k一直到数组末尾都是按照降序排列,将位置k+1到末尾reverse即可。
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int n = nums.size(), k, l;
for (k = n - 2; k >= 0; k--) {
if (nums[k] < nums[k + 1]) {
break;
}
}
if (k < 0) {
reverse(nums.begin(), nums.end());
} else {
for (l = n - 1; l > k; l--) {
if (nums[l] > nums[k]) {
break;
}
}
swap(nums[k], nums[l]);
reverse(nums.begin() + k + 1, nums.end());
}
}
};