【DFS】分书问题

Description

已知有n本书（从1～n编号）和n个人（从1～n编号），每个人都有一个自己喜爱的书的列表，现在请你编写一个程序，设计一种分书方案，使得每个人都能获得一本书，且这本书一定要在他的喜爱列表中。

Input

输入数据共若干行，第一行为一个正整数n（n <= 20），从第2行到第n+1行，每行有n个0或1组成，第k行表示编号为k-1的人对这n本书的喜好列表，0表示不喜欢，1表示喜欢。

Output

输出数据仅一个整数，表示符合条件的分配方案的总数。

Sample Input

5
00110
11001
01100
00010
01001

Sample Output

1

Code：

#include <iostream>

#define SIZE 25

using namespace std;

bool a[SIZE][SIZE], b[SIZE];
int res, n;

void dfs(int k) // 深搜过程
{
	if (k > n)
	{
		res++; // 结果总数 + 1
		return; // 这个层次不用再找了
	}
	for (int i = 1; i <= n; i++) // 枚举每一种情况
	{
		if ((!b[i]) && (a[k][i])) // 满足条件
		{
			b[i] = true; // 保存结果
			dfs(k + 1); // 递归
			b[i] = false; // 回溯
		}
	}
	
	return;
}

int main()
{
	int i, j;
	char c;
	
	cin >> n;
	for (i = 1; i <= n; i++) // 输入
	{
		for (j = 1; j <= n; j++)
		{
			cin >> c;
			if (c == '1')
			{
				a[i][j] = true;
			}
		}
	}
	
	dfs(1);
	
	cout << res << endl;
	
	return 0;
}