permutation 1 【HDU 6628】【暴力全排列】

题目链接

题目大意

我们定义1到n的全排列的“差排列”为n-1个数:p2-p1 , p3-p2 , … … … p_n - p_n-1
让你求出n的全排列的差排列字典序的第k大。

解题思路

因为k不大于10000所以可以采用爆搜的解法，当n不大于8的时候直接爆搜出它的全排列，如果n大于8的话，那么前面的数是固定n, 1, 2,… …所以只用求出后面的八位的全排列即可。
两个搜索其实可以并到一个dfs里面，我写的哟蒂娜麻烦… …

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
    int a[30];
};
node e[800005];
int vis[50],a[50],cnt,res,n,m;
int cmp(node x,node y)
{
    for(int i=1; i<m; i++)
    {
        if(x.a[i]-x.a[i-1]!=y.a[i]-y.a[i-1])
            return x.a[i]-x.a[i-1]<y.a[i]-y.a[i-1];
    }
}
void dfs(int pos)
{
    if(pos==m)
    {
        cnt++;
        for(int i=0; i<pos; i++)
        {
            // printf("%d ",a[i]);
            e[cnt].a[i]=a[i];
        }
       // printf("\n");

        return ;
    }
    for(int i=1; i<=n; i++)
    {
        if(vis[i]==0)
        {
            vis[i]=1;
            a[pos]=i;
            dfs(pos+1);
            vis[i]=0;
        }
    }
}
void dfs2(int pos)
{
    if(pos==m)
    {
        cnt++;
        for(int i=1; i<=pos; i++)
        {
            // printf("%d ",a[i]);
            e[cnt].a[i]=a[i];
        }
       // printf("\n");

        return ;
    }
    for(int i=1; i<=n; i++)
    {
        if(vis[i]==0)
        {
            vis[i]=1;
            a[pos]=i;
            dfs(pos+1);
            vis[i]=0;
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int k;
        cnt=0;
        scanf("%d %d",&n,&k);
        if(n<=8)
        {
            memset(vis,0,sizeof(vis));
            m=n;
            dfs(0);
            sort(e+1,e+1+cnt,cmp);
            for(int i=0; i<n; i++)
            {
                if(i)
                    printf(" ");
                printf("%d",e[k].a[i]);
            }
            printf("\n");
        }
        else
        {
            printf("%d",n);
            memset(vis,0,sizeof(vis));
            vis[n]=1;
            for(int i=1; i<=n-8-1; i++)
            {
                vis[i]=1;
                printf(" %d",i);
                a[0]=i;
            }
            m=9;
            dfs2(1);
            sort(e+1,e+1+cnt,cmp);
            for(int i=1; i<=8; i++)
            {
                printf(" %d",e[k].a[i]);
            }
            printf("\n");
        }
    }
    return 0;
}