3000米排名预测 DFS 全排列函数next_permutation

问题 1478: [蓝桥杯][算法提高VIP]3000米排名预测

时间限制: 1Sec 内存限制: 128MB 提交: 167 解决: 54

题目描述

3000米长跑时，围观党们兴高采烈地预测着  最后的排名。因为他们来自不同的班，对所有运动员不一定都了解，于是他们分别对自己了解的一些运动员的实力作出了评估，即对部分运动员做了相对排名的预  测，并且告诉了可怜留守的班长。因为无聊，于是他们就组团去打Dota去了。比赛结束后他们向班长询问最后的排名，但班长不记得了，只记得他们中哪些人的  预测是正确的，哪些人的预测是错误的。他们想知道比赛的排名可能是什么。

数据规模和约定
  1< =n< =10,  2< =c< =n,  1< =m< =10，保证数据合法，且答案中排名可能数不超过20000。对于一个排名序列，一个预测是正确的，当且仅当预测的排名的相对顺序  是排名序列的一个子序列。一个预测是错误的，当且仅当这个预测不正确。

输入

第一行两个整数n，  m，n为运动员数量，m为围观党数量。运动员编号从0到n-1。 
接下来m行，每行为一个围观党的相对排名预测。每行第一个数c表示他预测的人数，后面跟着c个0~n-1的不同的数，表示他预测的运动员相对排名，最后还有一个数，0表示这个预测是错误的，1表示是正确的。 

输出

第一行一个数k为有多少种排名的可能。 
下面k行，每行一个0~n-1的排列，为某一个可能的排名，相邻的数间用空格隔开。所有排名按字典序依次输出。 

样例输入

3 2 
2 0 1 1 
2 1 2 0 

样例输出

2
0 2 1 
2 0 1 

思路：

用全排列函数去试每一个可能的答案。然后利用给定的条件去判断这个全排列。

若给出的是错误的，那么若给出的答案是全排列的子序列，那么就是错的。

若给出的是正确的，那么若给出的答案不是全排列的子序列，那么答案就是错的。

代码1，利用全排列函数：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=20+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int dx[]= {-1,1,0,0};
int dy[]= {0,0,-1,1};
struct node
{
    int num;
    int order[maxn];
    int ok;
} a[maxn];
int n,m;
int ranks[maxn];
vector<int>ans[maxn*100];
int tot;
bool check()
{
    bool f1=1,f2=1;
    for(int i=0; i<m; i++) //m组
    {
        if(a[i].ok&&f1)
        {
            int j=0;
            for(int k=0; j<a[i].num&&k<n; k++)
            {
                if(a[i].order[j]==ranks[k])
                {
                    j++;
                }
            }
            if(j!=a[i].num)
                f1=0;
        }
        if(!a[i].ok&&f2)
        {
            int j=0;
            for(int k=0; j<a[i].num&&k<n; k++)
            {
                if(a[i].order[j]==ranks[k])
                {
                    j++;
                }
            }
            if(j==a[i].num)
                f2=0;
        }
        if(!f1||!f2)break;
    }
    if(f1&&f2)return 1;
    return 0;
}
void dfs()
{
    for(int i=0; i<n; i++)
        ranks[i]=i;
    do
    {
        if(check())
        {
            for(int i=0; i<n; i++)
            {
                ans[tot].push_back(ranks[i]);
            }
            tot++;
        }
    }
    while(next_permutation(ranks,ranks+n)); //全排列函数
}
int main()
{
    cin>>n>>m;
    for(int i=0; i<m; i++)
    {
        cin>>a[i].num;
        for(int j=0; j<a[i].num; j++)
        {
            cin>>a[i].order[j];
        }
        cin>>a[i].ok;
    }
    dfs();
    cout<<tot<<endl;
    for(int i=0;i<tot;i++)
    {
        int s=ans[i].size();
        for(int j=0;j<s;j++)
        {
            cout<<ans[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}

代码2，利用DFS手写全排列：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=20+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int dx[]= {-1,1,0,0};
int dy[]= {0,0,-1,1};
struct node
{
    int num;
    int order[maxn];
    int ok;
} a[maxn];
int n,m;
int ranks[maxn];
vector<int>ans[maxn*1000];
int tot;
bool check()
{
    bool f1=1,f2=1;
    for(int i=0; i<m; i++) //m×é
    {
        if(a[i].ok&&f1)
        {
            int j=0;
            for(int k=0; j<a[i].num&&k<n; k++)
            {
                if(a[i].order[j]==ranks[k])
                {
                    j++;
                }
            }
            if(j!=a[i].num)
                f1=0;
        }
        if(!a[i].ok&&f2)
        {
            int j=0;
            for(int k=0; j<a[i].num&&k<n; k++)
            {
                if(a[i].order[j]==ranks[k])
                {
                    j++;
                }
            }
            if(j==a[i].num)
                f2=0;
        }
        if(!f1||!f2)
            break;
    }
    if(f1&&f2)
        return 1;
    return 0;
}
bool vis[maxn];
void dfs(int x)
{
    if(x==n)
    {
        if(check())
        {
            for(int i=0; i<n; i++)
            {
                ans[tot].push_back(ranks[i]);
            }
            tot++;
        }
        return;
    }
    for(int i=0; i<n; i++)
    {
        if(!vis[i])
        {
            vis[i]=1;
            ranks[x]=i;
            dfs(x+1);
            vis[i]=0;
        }
    }
    return;
}
int main()
{
    cin>>n>>m;
    for(int i=0; i<m; i++)
    {
        cin>>a[i].num;
        for(int j=0; j<a[i].num; j++)
        {
            cin>>a[i].order[j];
        }
        cin>>a[i].ok;
    }
    for(int i=0; i<n; i++)
        ranks[i]=i;
    dfs(0);
    cout<<tot<<endl;
    for(int i=0; i<tot; i++)
    {
        int s=ans[i].size();
        for(int j=0; j<s; j++)
        {
            cout<<ans[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}