AcWing 420. 火星人

原题链接

考察:STL(?)

思路:

       最简单的做法:next_permutation 全排列.

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 10010;
int a[N];
int main()
{
    int n,cnt;
    scanf("%d%d",&n,&cnt);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    while(cnt--)
        next_permutation(a+1,a+n+1);
    for(int i=1;i<=n;i++) printf("%d ",a[i]);
    return 0;
}

      做法二:模拟next_permutation

      思路:模拟发现,字典序的全排列,由第i个变到第i+1个是将第k位数字变得大一点,这个第k位就是从尾到头第一个不是升序的数字.我们要做的是将a[k]变成a[k]+1,k位后的数字需要让它变成字典序最小,也就是reverse

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 10010;
int a[N];
int main()
{
    int n,cnt;
    scanf("%d%d",&n,&cnt);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    while(cnt--)
    {
        int k = n-1;
        while(a[k]>a[k+1]) k--;
        int t = k++;
        while(a[t]<a[k]) k++;
        swap(a[t],a[k-1]);
        reverse(a+t+1,a+n+1);
        //printf("\n");
    }
    for(int i=1;i<=n;i++) printf("%d ",a[i]);
    return 0;
}