題意:
給出一串數字,求該排列的下一個排列。如果該排列爲字典序最大排列,則輸出字典序最小排列。
思路:
首先C++裏面有求下一個排列的函數next_permutation,該函數返回0即表示當前排列字典序最大。
如果要自己動手實現,那麼就考慮“如何找到比當前排列大1的排列”。
假設排列A(aaaXddd)比排列B(aaaYfff)大1,a部分爲兩排列相同部分,X與Y表示最靠左邊不同的那一位。那麼,X>Y且X一定在B中的f部分且f中數字一定是遞減排序的且d一定是遞增排序的。
代碼:
class Solution {
public:
void nextPermutation(vector<int> &nums) {
int idx = findPos(nums);
if (idx == -1) {
reverse(nums.begin(), nums.end());
} else {
reverse(nums.begin() + idx + 1, nums.end());
for (int i = idx + 1; i < nums.size(); ++i) {
if (nums[idx] < nums[i]) {
swap(nums[idx], nums[i]);
break;
}
}
}
}
private:
int findPos(vector<int> &nums) {
for (int i = nums.size() - 2; i >= 0; --i) {
if (nums[i] < nums[i + 1]) {
return i;
}
}
return -1;
}
};
/**
* 偷懶版本……
*/
class Solution {
public:
void nextPermutation(vector<int> &nums) {
if (!next_permutation(nums.begin(), nums.end())) {
sort(nums.begin(), nums.end());
}
}
};