class Solution {
public:
void nextPermutation(vector<int>& nums) {
int index = -1;
int tmp;
for (int i = nums.size()-1; i > 0; --i) {
if (nums[i-1] < nums[i]) {
index = i-1;
break;
}
}
if (index == -1) {
sort(nums.begin(), nums.end(), [](int a, int b){return a<b;});
} else {
for (int i = nums.size()-1; i > index; --i) {
if (nums[i] > nums[index]) {
tmp = nums[i];
nums[i] = nums[index];
nums[index] = tmp;
break;
}
}
sort(nums.begin()+index+1, nums.end(), [](int a, int b){return a<b;});
}
}
};
思路
根据排列的性质,想按照字典序得到下一个排列,分为两步:
第一步,从后往前,找到第一个不满足递增性质的元素;
第二步,将该元素和后面第一个大于该元素的元素交换,然后后面的元素按照从小到大排列,即得下一个排列;
要点
思路正确性证明:
按照选元素的方法,待交换元素的后面元素一定是逆序排列且存在大于该元素的值;从而下一个排列必然以其中下一个大于该元素的值取代该元素的位置开始,后面只要是顺序排列,即是该情况下最小排列,从而整个就是最小排列;
sort()函数使用传入两个迭代器,第三个参数表示元素排序的方法,使用Lambda表达式;