【基础算法】next_permutation的手动实现

求下一个序列的考虑方法：
尽量保证地位不变，变高位。所以可以从后往前去考虑

视频讲解：LeetCode 31. 下一个排列

模板题：

31. 下一个排列

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int k = nums.size() - 1; // k为最后一个数的下标（下标从0开始）
        //从后往前找，找到第一个逆序的位置
        while(k > 0 && nums[k - 1] >= nums[k]) k -- ;
        //停下来后，nums[k]是从后往前第一个逆序对中较大的数
        if(k <= 0) //整个序列逆序，直接翻转
        {
            reverse(nums.begin(), nums.end());
        }
        else
        {
            int t = k;
            while(t < nums.size() && nums[t] > nums[k - 1]) t ++ ;
            //nums[k - 1]要与nums[k]后面大于等于nums[k - 1]的最小的数交换
            swap(nums[t - 1], nums[k - 1]);
            reverse(nums.begin() + k, nums.end());
        }
    }
};

420. 火星人

下标从0开始

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1e4 + 10;

int n, m;
int w[N];

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 0; i < n; i ++ ) scanf("%d", &w[i]);
    while(m -- )
    {
        int k = n;
        while(w[k - 1] >= w[k]) k -- ; //找到第一对逆序数，w[k]为逆序对中较大的数，这里等号写不写都可
        int t = k;
        while(t < n && w[t] > w[k - 1]) t ++ ;
        swap(w[t - 1], w[k - 1]);
        reverse(w + k, w + n);
    }
    for(int i = 0; i < n; i ++ ) printf("%d ", w[i]);
    return 0;
}

下标从1开始

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1e4 + 10;

int n, m;
int w[N];

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
    while(m -- )
    {
        int k = n;
        while(w[k - 1] > w[k]) k -- ; 
        int t = k;
        while(w[t + 1] > w[k - 1]) t ++ ;
        swap(w[k - 1], w[t]);
        reverse(w + k, w + n + 1);
    }
    for(int i = 1; i <= n; i ++ ) printf("%d ", w[i]);
    return 0;
}

作者：Once.
链接：https://www.acwing.com/activity/content/code/content/3396797/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

next_permutation写法

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1e4 + 10;

int n, m;
int w[N];

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
    while(m -- )
    {
        next_permutation(w + 1, w + n + 1);
    }
    for(int i = 1; i <= n; i ++ ) printf("%d ", w[i]);
    return 0;
}

作者：Once.
链接：https://www.acwing.com/activity/content/code/content/3396797/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。