- LeetCode(力扣)
题目:
31. 下一个排列
难度:中等 🌟🌟🌟
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
题解:
首先,需要理解 字典序 。
举个例子, [ 1 , 2 , 3 ] [1,2,3] [1,2,3] ,其字典序排列如下:
[ 1 , 2 , 3 ] [ 1 , 3 , 2 ] [ 2 , 1 , 3 ] [ 2 , 3 , 1 ] [ 3 , 1 , 2 ] [ 3 , 2 , 1 ] [1,2,3][1,3,2][2,1,3][2,3,1][3,1,2][3,2,1] [1,2,3][1,3,2][2,1,3][2,3,1][3,1,2][3,2,1]
比如, [ 2 , 3 , 1 ] [2,3,1] [2,3,1] 的下一个为 [ 3 , 2 , 1 ] [3,2,1] [3,2,1] ,特别的对于 [ 3 , 2 , 1 ] [3,2,1] [3,2,1] 的下一个为 [ 1 , 2 , 3 ] [1,2,3] [1,2,3]
算法流程,假设数组长度为 n n n 的数据 a a a :
- 首先从最右边往左查找第一组满足 a [ i + 1 ] > a [ i ] a[i+1]>a[i] a[i+1]>a[i] ,记录当前的位置 F i r s t i n d e x = i Firstindex=i Firstindex=i,此时,可以知道, [ i + 1 , n − 1 ] [i+1,n-1] [i+1,n−1] 必然为递减序列 (如果找不到满足条件的组,那么说明该序列已经是最大的序列,直接翻转整个序列即可)
- 然后在 [ i + 1 , n − 1 ] [i+1,n-1] [i+1,n−1] 中从右往后找到第一个满足 a [ j ] > a [ F i r s t i n d e x ] a[j]>a[Firstindex] a[j]>a[Firstindex] 的元素,记录当前的位置 S e c o n d i n d e x = j Secondindex=j Secondindex=j
- 此时,交换 a [ F i r s t i n d e x ] , a [ S e c o n d i n d e x ] a[Firstindex],a[Secondindex] a[Firstindex],a[Secondindex] ,可以证明 [ i + 1 , n − 1 ] [i+1,n-1] [i+1,n−1] 必然为降序。最后,我们让区间 [ i + 1 , n − 1 ] [i+1,n-1] [i+1,n−1] 变为升序即完成。
程序实现:
-
Python
class Solution: def nextPermutation(self, nums: List[int]) -> None: """ Do not return anything, modify nums in-place instead. """ def reverse(a,i,j): while i<j: a[i],a[j] = a[j],a[i] i += 1 j -= 1 n = len(nums) first_idx,second_idx = -1,-1 for i in range(n-1,0,-1): if(nums[i] > nums[i-1]): first_idx = i-1 break if(first_idx == -1): reverse(nums,0,n-1) return for i in range(n-1,first_idx,-1): if(nums[i] > nums[first_idx]): second_idx = i break nums[first_idx],nums[second_idx] = nums[second_idx],nums[first_idx] reverse(nums,first_idx+1,n-1) return
-
c++
class Solution { public: void nextPermutation(vector<int>& nums) { int i=nums.size()-2; while(i>=0 && nums[i]>=nums[i+1]) { i--; } if(i>=0) { int j=nums.size()-1; while(j>=0 && nums[j]<=nums[i]) j--; swap(nums[i], nums[j]); } reverse(nums.begin()+i+1,nums.end()); } };
PS: 当然 c++ 也有强大的标准库函数 next_permutation
复杂度分析:
时间复杂度: O ( N ) O(N) O(N) ,其中 N N N 为给定序列的长度。我们至多只需要扫描两次序列,以及进行一次反转操作。
空间复杂度: O ( 1 ) O(1) O(1) ,只需要常数的空间存放若干变量。