Joyce' Blog
0%

next-permutation

发表于

题目描述

实现获取下一个排列的函数，算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。

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

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

题解

要找到需要替换的关键位置，即nums[i]<nums[i+1]，然后找出后半段中最小的 大于nums[i] 的数，剩下的只需从小到大重新放入。

时间上还是不错的，空间做的不够好。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
void nextPermutation(vector<int>& nums) {
	int size = nums.size();
	if (size <= 1)return;
	int* tmp = new int[size];
	int i;
	tmp[size - 1] = nums.back();
	for (i = size - 1; i > 0; i--)
	{
		int j = nums[i];
		int k = nums[i - 1];
		tmp[i-1] = k;
		if (j > k)
		{
			sort(tmp + i - 1, tmp + size);
			for (int x = size; x >= i; x--)
				nums.pop_back();
			int x;
			for (x = size - 2; x >=0; x--)
			{
				if (tmp[x] == k)
				{
					k = tmp[x + 1];
					nums.push_back(k);
					break;
				}
			}
			int y = x + 1;
			for (x = i - 1; x < size; x++)
				if (x!=y)
					nums.push_back(tmp[x]);
			break;
		}
	}
	if (i == 0)
	{
		nums.clear();
		for (i = size-1; i >=0; i--)
			nums.push_back(tmp[i]);
	}
}