P1657 选书（变形dfs）

P1657 选书

题目描述

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

输入格式

第1行：一个数x

第2行~第1+x行：每行两个数，表示ai喜欢的书的序号

输出格式

只有一个数：总方案数total。

输入输出样例

输入 #1

5
1 3
4 5
2 5
1 4
3 5

输出 #1

2

说明/提示

所有数据：x<=20

（世界上最难出数据的题目，没有之一……）

我的思路

本人的思路就是判断是否选了x本书并且这些书没有重复，满足这些条件方案就+1。

AC代码

#include<bits/stdc++.h>
using namespace std;
int a[21][2], x, d[21], ans;
void dfs(int i) {
  if (i > x) {
    ans++;//当i超过x本书，方案+1 
    return;
  }
  for (int j = 0; j <= 1; j++) 
    if (!d[a[i][j]]) { //判断此书是否被选过 
      d[a[i][j]] = 1;//标记此书已被选 
      dfs(i + 1);//继续选下一本 
      d[a[i][j]] = 0;//回溯，考虑其他情况 
    }
  return;
} 
int main() {
  cin >> x;
  if (x == 0) {//这里需要特判，因为如果x = 0，ans输出1 
    cout << 0;
    return 0;
  }
  for (int i = 1; i <= x; i++)
    cin >> a[i][0] >> a[i][1];
  dfs(1);
  cout << ans;
  return 0;
}