题目链接:leetcode31
题目大意
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解题思路
next_permutation
利用『algorithm』包下的 API ,当返回为false时说明完全逆序,反转一下即可。
思维
观察一组长度为 n 的序列,我们记它为 P(n) 。关于问题要求 P(n) 的下一个排列,我们记它的下一个排列为 F(n) 。根据 F(n) 和 P(n),我们可以发现,两者在非完全逆序的情况下,存在若干个相同的前缀,我们记这个数量为 k 。那么,对于 P(n) 转化为 F(n) 的问题变成了更加简单的 P(n-k) 转化为 F(n-k) 的子问题,也就是 n-k 个后缀的问题,我们记 s=n-k。
对于 P(s) 我们可以从后往前探测,如果发现一个位置不完全逆序这个点就是我们的操作点,可以把后方最后一个比它大的和它交换,此时发现逆序数+1,且这个位置后面的序列单调性不变,但显然要把后面的数字转成递增,反转一下即可。为什么要这样做呢?因为从后面交换到一个最接近当前数字并比它大的时候,这段序列实际上已经『革新』了,就好比我们已经确定了 F(s) 的开头,但对于后续的 P(s-1) 部分应该转成逆序数为0的序列,例如数字中 99+1 进位的时候,高位扩大,低位应该回归最小一个道理。由于后续一定是非递减的,所以反转一下恰好可以消除逆序。
以上是我对这个算法的个人理解,不一定正确。不过没找到这个算法严格的数学证明,只有一份类似于归纳的证明,可以蛮看看 证明
时间复杂度 O ( n ) O(n) O(n),空间复杂度 O ( 1 ) O(1) O(1)。
代码实现
/*
* @lc app=leetcode.cn id=31 lang=cpp
*
* [31] 下一个排列
*/
// @lc code=start
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if (nums.size() < 2) {
return ;
}
vector<int>::iterator pre = nums.end() - 1, suf, first_bigger;
while (true) {
suf = pre;
pre--;
if (*pre < *suf) {
first_bigger = nums.end() - 1;
while(*pre >= *first_bigger) first_bigger--;
swap(*first_bigger, *pre);
reverse(suf, nums.end());
break;
} else if (pre == nums.begin()) {
reverse(nums.begin(), nums.end());
break;
}
}
}
};
// @lc code=end