AcWing 420. 火星人

y总讲得很好，学到很多所以安利一下

转载自Acwing yxc

算法
(贪心,全排列) O(nm)O(nm)
这道题目可以直接用next_permutation函数来做。

这里我们考虑一下next_permutation函数的原理，然后手动实现一遍。

对于给定的某个排列，我们想求出比它大的最小的排列。
可以从后往前遍历这个排列，找到第一个可以让排列的字典序变大的位置。

只有当序列单调下降时，它才不存在更大的排列，因此我们要找的位置就是第一次出现 ak−1<akak−1<ak 的位置。

那么此时将 ak−1ak−1 变成比它大的最小数，然后将剩余部分从小到大排序，得到的排列就是比原排列大的最小排列了。

这里有个小优化：

由于 ak−1ak−1 后面的部分已经从大到小排好序，因此只需将其翻转，就可以得到从小到大排序的结果了，不需要使用 sort函数，时间效率可以降到线性。
时间复杂度
一共求 mm 次next_permutation，每次需要 o(n)o(n) 的时间，因此总时间复杂度是 O(nm)O(nm)。

C++ 代码

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

using namespace std;

const int N = 10010;

int n, m;
int q[N];

int main()
{
    
    
    scanf("%d%d", &n, &m);

    for (int i = 0; i < n; i ++ ) scanf("%d", &q[i]);

    while (m -- )
    {
    
    
        int k = n - 1;
        while (q[k - 1] > q[k]) k -- ;
        k -- ;
        int t = k;
        while (t + 1 < n && q[t + 1] > q[k]) t ++ ;
        swap(q[t], q[k]);
        reverse(q + k + 1, q + n);
    }

    for (int i = 0; i < n; i ++ ) printf("%d ", q[i]);

    return 0;
}

作者：yxc
链接：https://www.acwing.com/solution/content/4504/
来源：AcWing