1. 程式人生 > 實用技巧 >31.下一個排列

31.下一個排列

實現獲取下一個排列的函式,演算法需要將給定數字序列重新排列成字典序中下一個更大的排列。

如果不存在下一個更大的排列,則將數字重新排列成最小的排列(即升序排列)。

必須原地修改,只允許使用額外常數空間。

以下是一些例子,輸入位於左側列,其相應輸出位於右側列。

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

演算法:

(1)從後往前遍歷,找到第一個比後面小的一個數,記錄其下標為i。
若找不到則表明當前排列為全排列最後一個排列,反轉整個排列即可。
(2)從後往前遍歷,找到第一個大於nums[i]的數,記錄其下標為j。
(3)swap(nums[i],nums[j]);
此時i後面仍然是區域性全排列的最後一個排列。
(4)reverse(nums.begin()+i+1,nums.end()),即反轉下標i後面的所有數。
即將i後面的排列換為區域性全排列的第一個排列。

舉例:
1 2 7 4 3 1
1 2 7 4 3 1
1 3 7 4 2 1
1 3 1 2 4 7

void nextPermutation(vector<int>& nums) {
	int i, j, n = nums.size();
	for (int i = n - 2; i >= 0; i--) {
		if (nums[i + 1] > nums[i]) {
			//確定i
			for (int j = n - 1; j > i; --j) {
				//確定j
				if (nums[j] > nums[i]) break;
			}
			swap(nums[i], nums[j]);
			reverse(nums.begin() + i + 1, nums.end());
			return;
		}
	}
	//不存在下一個更大的排列,將數字重新排列為最小的排列
	reverse(nums.begin(), nums.end());
}

leetcode原題連結:https://leetcode-cn.com/problems/next-permutation