LeetCode C++9-下一个排列

题目描述

    给你一个整数数组nums，找出nums的下一个排列。

    要求：必须原地修改，只允许使用额外常数空间。

举例

示例1：nums=[1,2,3] ==> 输出：[1,3,2]

示例2：nums=[3,2,1] ==> 输出：[1,2,3]

示例3：nums=[1,1,5]==> 输出：[1,5,1]

解题方法：假设数组的大小为n，先从右向左寻找第一个降序的数字，假设其下标为i，再在区间[i+1， n-1]从右向左寻找第一个比nums[i]大的数，与nums[i]交换。交换后，再将[i+1, n-1]的元素翻转即可。

代码

#include <iostream>
#include <algorithm>
#include <vector>

void next_permutation(std::vector<int>& nums) {
	int n = nums.size();
	int right = n - 1; // 记录从右向左最后一个升序的元素的下标
	while (right > 0) {
		if (nums[right - 1] < nums[right]) {
			break;
		}
		right--;
	}
	if (right > 0) { //从右向左存在逆序的数字
		for (int i = n - 1; i >= right; i--) {
			if (nums[i] > nums[right - 1]) {
				std::swap(nums[i], nums[right - 1]);
				break;
			}
		}
	}
	std::reverse(nums.begin() + right, nums.end());
}

void print_nums(const std::vector<int>& nums) {
    for (int value : nums) {
		std::cout << value << " ";
	}
	std::cout << std::endl;
}

int main()
{
	// case1: nums=[1,2,3] ==> [1,3,2]
	std::vector<int> nums = {1, 2, 3};
	std::cout << "==case1==" << std::endl;
	print_nums(nums);
	next_permutation(nums);
	print_nums(nums);
	// case2: nums=[3,2,1] ==> [1,2,3]
	std::cout << "==case2==" << std::endl;
	nums = {3, 2, 1};
	print_nums(nums);
	next_permutation(nums);
	print_nums(nums);
	// case3: nums=[1,1,5] ==> [1,1,5]
    std::cout << "==case3==" << std::endl;
	nums = {1, 1, 5};
	print_nums(nums);
	next_permutation(nums);
	print_nums(nums);
    return 0;
}

代码输出如下：

==case1==
1 2 3 
1 3 2 
==case2==
3 2 1 
1 2 3 
==case3==
1 1 5 
1 5 1