选书(DFS)

题面(from luogu)
选书
学校放寒假时,信息学奥赛辅导老师有1,2,3……x本书,要分给参加培训的x个人,每人只能选一本书,但是每人有两本喜欢的书。老师事先让每个人将自己喜欢的书填写在一张表上。然后根据他们填写的表来分配书本,希望设计一个程序帮助老师求出所有可能的分配方案,使每个学生都满意。

输入格式:
第1行:一个数x
第2行~第1+x行:每行两个数,表示ai喜欢的书的序号
输出格式:
只有一个数:总方案数total。
样例.in
5
1 3
4 5
2 5
1 4
3 5
样例.out
2

题目分析
同样的DFS……
我们可以先按照题目的要求生成全排列(但不用存下来),每生成完一次计数器就++

大体框架

void search(int k)
{
    if (长度符合n) 
            ans++;
            else
                {
                    for (int i = 1; i <= n; i++)
                        {
                            if (没有人要 && 有人喜欢)
                                {
                                    打标记;
                                    向前一步搜索
                                    回溯;     
                                }                   
                        }
                }
}

代码

#include <bits/stdc++.h>
using namespace std;

int vis[23],cnt[23][23];
int n,ans; 

void search(int k)
{
    if (k == n+1) //生成好了
            ans++;   
            else   //反之,继续找
                {
                    for (int i = 1; i <= n; i++)
                        {
                            if (vis[i]==1 && cnt[k][i]==1)   //这本书目前还没有人要并且有人喜欢这本书
                                {
                                    vis[i]=0;   //给这本书打上标记
                                    search(k+1);   //向前一步搜索
                                    vis[i]=1;   //回溯            
                                }                   
                        }
                }
}

int main()
{
    int t1,t2;

    cin>>n;  //输入

    if (n == 0)   //坑点,因此WA了一次
        {
            cout<<0;
            return 0;
        }

    for (int i = 1; i <= n; i++)
        {
            cin>>t1>>t2;
            cnt[i][t1]=1;  //给有人喜欢的书打标记
            cnt[i][t2]=1;
            vis[i]=1;  //初始化
        }

    search(1);  //开始搜索

    cout<<ans;  //输出
    return 0;  //完美的结束程序
}
                                           **蒟蒻新星c_uizrp_dzjopkl原创**

猜你喜欢

转载自blog.csdn.net/c_uizrp_dzjopkl/article/details/81781568