假期的宿舍 - C++

假期的宿舍

题目描述
学校放假了 · · · · · · 有些同学回家了，而有些同学则有以前的好朋友来探访，那么住宿就是一个问题。
比如 A 和 B 都是学校的学生，A 要回家，而 C 来看B，C 与 A 不认识。我们假设每个人只能睡和自己直接认识的人的床。那么一个解决方案就是 B 睡 A 的床而 C 睡 B 的床。而实际情况可能非常复杂，有的人可能认识好多在校学生，在校学生之间也不一定都互相认识。
我们已知一共有 n 个人，并且知道其中每个人是不是本校学生，也知道每个本校学生是否回家。问是否存在一个方案使得所有不回家的本校学生和来看他们的其他人都有地方住。
输入格式
第一行一个数 T 表示数据组数。接下来 T 组数据，每组数据第一行一个数n 表示涉及到的总人数。
接下来一行 n 个数，第 i 个数表示第 i 个人是否是在校学生 (0 表示不是，1 表示是)。再接下来一行 n 个数，第 i 个数表示第 i 个人是否回家 (0 表示不回家，1 表示回家，注意如果第 i 个人不是在校学生，那么这个位置上的数是一个随机的数，你应该在读入以后忽略它)。
接下来 n 行每行 n 个数，第 i 行第 j 个数表示 i 和 j 是否认识 (1 表示认识，0 表示不认识，第 i 行 i 个的值为 0，但是显然自己还是可以睡自己的床)，认识的关系是相互的。
输出格式
对于每组数据，如果存在一个方案则输出 “^_”(不含引号) 否则输出“T_T”(不含引号)。(注意输出的都是半角字符，即三个符号的 ASCII 码分别为94,84,95)
输入输出样例
输入 #1
1
3
1 1 0
0 1 0
0 1 1
1 0 0
1 0 0
输出 #1
^_

这题显然是二分图匹配问题。又多了些限制条件。注意的是这里不是人和人匹配，是人和床匹配。是不回家的学生（包括外校和本校）匹配本校学生的床（因为只有本校学生才有床）。
C++代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int stu[10001],ho[10001],n;
int e[10001][10001];
int book[10001];
int match[10001];
bool dfs(int u)
{
    for(int i = 1; i <= n; i ++)
    {
       //只要是本校学生都有床位 
        if(book[i] == 0 &&  stu[i] == 1 && e[u][i] == 1)
   {
    book[i] = 1;
        if(match[i] == 0 || dfs(match[i]))
   {
      match[i] = u;
      return 1;  
   } 
    }     
    } 
    return false;
}
bool check()
{
     for(int i = 1; i <= n; i ++)
     {
          for(int j = 1; j <= n; j ++)
          book[j] = 0;
          //只要学生不回家就匹配 无论是外校还是本校 
          if((stu[i] == 0 || ho[i] == 0) && dfs(i) == false)
     return false; 
     }
     return true; 
} 
int main()
{
 int t;
 scanf("%d",&t);
 while(t --)
 {
     cin >> n;
     memset(stu,0,sizeof(stu));
     memset(ho,0,sizeof(ho));
     memset(match,0,sizeof(match)); 
     for(int i = 0; i <= n; i ++)
     for(int j = 0; j <= n; j ++)
     e[i][j] = 0;
     
     for(int i = 1; i <= n; i ++)
     {
        scanf("%d",&stu[i]);     
     } 
     for(int i = 1; i <= n; i ++)
     {
        scanf("%d",&ho[i]);
        //如果是外校学生 则没有床位 所以为1 
        if(stu[i] == 0)
        ho[i] = 1;
     }
     
     for(int i = 1; i <= n; i ++)
     for(int j = 1; j <= n; j ++)
      {
         scanf("%d",&e[i][j]);
         //自己能和自己的床匹配 
         if(stu[i] == 1)
         e[i][i] = 1;
      }
      //
     if(check() == true)
     {
      cout << "^_^" << endl;
     }else
     cout << "T_T" << endl;
 }
      return 0; 
 }