题目描述
给出一个数组,找出在排列组合的字典序中,它的下一个组合。如果不存在下一个的话,返回第一个。
如
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解题思想
如果不存在下一个组合的话,说明它是字典序最大的组合,这是数组是逆序的,将数组调换即可。
如果存在下一个组合,那么数组一定不是逆序的,也就有某一个元素,大于它的前一个元素,找到这样的位置。比如对于数组 [1, 2, 5, 8, 7]
,元素 8
就大于元素 5
,找到这样的位置以后,将 5
与其后的最小的一个并且大于 5
元素交换,这样可以保证交换完以后的字典序比现在大。比如这里是和 7
交换。交换完成以后,将 5
之后剩下的数组排序即可。这样的话,就是字典序比之前的字典序大的排列中最小的一个排列,也就是要返回的结果。
代码
class Solution {
public:
void nextPermutation(vector<int>& nums) {
// 1 2 5 8 7 ->
// 1 2 7 5 8
for (int i = nums.size() - 1; i > 0; i --){
if (nums[i] > nums[i - 1]){
// find next larger index
int min = INT_MAX, j, min_index;
for (j = nums.size() - 1; j >= i; j --){
if (nums[j] > nums[i - 1] && nums[j] < min){
min = nums[j];
min_index = j;
}
}
swap(nums[i - 1], nums[min_index]);
sort(nums.begin() + i, nums.end());
return;
}
}
for (int i = 0; i < nums.size() / 2; i ++){
swap(nums[i], nums[nums.size() - 1 - i]);
}
}
};