class Solution {
public:
void nextPermutation(vector<int>& nums) {
int n = nums.size();
if (n <= 1) return;
int i = n - 2, j = n - 1, k = n - 1;
while (i >= 0 && nums[i] >= nums[j]) {
i--;
j--;
}
if (i >= 0) {
while (nums[i] >= nums[k]) {
k--;
}
swap(nums[i], nums[k]);
}
for (int m = j, n = nums.size() - 1; m < n; ++m, --n) {
swap(nums[m], nums[n]);
}
}
};
代码(二刷看解析 2024年4月1日)
class Solution {
public:
void nextPermutation(vector<int>& nums) {
// 一个指针指向最后一个元素,
//一个指针往前找,找到第一个小于最后一个元素的指针后交换位置
int left = nums.size() - 1;
// 单调递增栈,若遇到比栈顶小的,开始出栈,直到找到最小的
stack<int> stk;
while (left >= 0) {
//栈空入栈
if (stk.empty() || nums[left] >= nums[stk.top()])
stk.push(left);
//栈非空,对比栈顶元素和当前元素
else {
int right = 0;
while (!stk.empty() && nums[left] < nums[stk.top()]) {
right = stk.top();
stk.pop();
}
swap(nums[right], nums[left]);
sort(nums.begin() + left + 1, nums.end());
return;
}
left--;
}
//若遍历结束都没找到,返回nums的倒序
reverse(nums.begin(), nums.end());
}
};
代码(三刷看解析 2024年9月5日)
class Solution {
public:
// 比如有个数10321,那最优的方案就是把最后的1和0交换,得到11320,然后把321翻转一下,得到
// 11023,这样就是比它大的数中最小的了。
// 代码思想就是:从后往前,找到第一个满足nums[i]<nums[i+1]的数,此时这个位置就是可以被改变的位置,相当于上面的0
// 然后i+1->end肯定是递减的,然后从后往前,找到第一个比nums[i]大的数,此时这个数就是比nums[i]大的数中,最小的那个,相当于上面的1
// 然后把这两个数交换一下,再把i+1后面的数翻转一下即可
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[i] >= nums[j]) {
j--;
}
swap(nums[i], nums[j]);
}
reverse(nums.begin() + i + 1, nums.end());
}
};