UPC（25）问题 A: 分书问题

时间限制: 1 Sec 内存限制: 128 MB

题目描述
已知有n本书（从1～n编号）和n个人（从1～n编号），每个人都有一个自己喜爱的书的列表，现在请你编写一个程序，设计一种分书方案，使得每个人都能获得一本书，且这本书一定要在他的喜爱列表中。
输入
输入数据共若干行，第一行为一个正整数n（n <= 20），从第2行到第n+1行，每行有n个0或1组成，第k行表示编号为k-1的人对这n本书的喜好列表，0表示不喜欢，1表示喜欢。
输出
输出数据仅一个整数，表示符合条件的分配方案的总数。
样例输入
5
00110
11001
01100
00010
01001
样例输出
1
解题思路：深度搜索，从第一个人（第一行）开始搜索（dfs），判断对应的“书”是否被借，进行到第 n个人的时候，ans++。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <math.h>
#include <vector>
#include <queue>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
int n,i,j,ans=0;
int vis[25];//记录第i本书是否被借
int a[25][25];
char str[25];//用于输入
void dfs(int m,int k)
{
    if(m>n||m<1)//递归终止条件
        return ;
    for(int i=1;i<=n;i++)
    {
        if(a[m+1][i]==1&&vis[i]==0)
        {
            vis[i]=1;
            if(m+1==n)//
                ans++;
            else dfs(m+1,i);//没有到达n，继续搜索
            vis[i]=0;
        }
    }
}
int main()
{
    scanf("%d",&n);
    for(i=1; i<=n; i++)
    {
        scanf("%s",str);//一个输入的小技巧
        for(j=1; j<=n; j++)
            a[i][j]=str[j-1]-'0';//字符串转化
    }
    for(i=1; i<=n; i++)
    {
      memset(vis,0,sizeof(vis));//每次遍历初始化
      if(a[1][i]==1)//判断第一个人是否借到书，进而进行搜索，如果false，直接不进行搜索
      {
          vis[i]=1;
          dfs(1,i);
      }
    }
    printf("%d",ans);
    return 0;
}

再接再厉！