洛谷 P2055 【假期的宿舍】

• 题库 ：洛谷
• 题号 ：2055
• 题目 ：假期的宿舍
• link ：https://www.luogu.org/problem/P2055

---

首先明确一下：校内的每个学生都有一张床（只是校内的有）

思路 ：分析题目发现是求所有在校学生能否全部有床睡（注意：只需在校学生有床睡，及不回家的；这个床可以是自己的，也可以是朋友的）；于是，这道题我们可以把它想象成一个二分图，如样例图：

a -> b表示a可以睡b的床（注意：虽然样例中没有2 -> 2的边，但你仔细读题就会发现题目中的原话---）

由于题目里给的是一个邻接矩阵，我们便可以无视蓝点所连出去的边，只用红点去搜就行了（代码里写了也不要紧）

代码 ：

 #include <bits/stdc++.h>
 #define INF 0x3f3f3f3f
 using namespace std;
 int T, n, head[], num, vis[], choose[], q[], w[], ans, cnt;
 struct node
 {
     int next, to;
 }stu[];
 inline void add(int x, int y)//链式前向星
 {
     stu[++num].next = head[x];
     stu[num].to = y;
     head[x] = num;
     return;
 }
 inline int dfs(int u)//二分图匈牙利算法模板
 {
     for(register int i = head[u]; i; i = stu[i].next)
     {
         int k = stu[i].to;
         if(vis[k])
         {
             continue;
         }
         vis[k] = ;
         if(!choose[k] || dfs(choose[k]))
         {
             choose[k] = u;
             return ;
         }
     }
     return ;
 }
 signed main()
 {
     scanf("%d", &T);
     while(T--)//多组数据
     {
         memset(choose, , sizeof(choose));//初始化
         memset(head, , sizeof(head));
         num = ;
         ans = ;
         cnt = ;
         scanf("%d", &n);
         for(register int i = ; i <= n; ++i)
         {
             scanf("%d", &q[i]);
         }
         for(register int i = ; i <= n; ++i)
         {
             scanf("%d", &w[i]);
             if(q[i] && !w[i])//如果他是本校学生&&他不回家
             {
                 add(i, i);//自己可以睡自己的床
             }
         }
         for(register int i = ; i <= n; ++i)
         {
             for(register int j = , x; j <= n; ++j)
             {
                 scanf("%d", &x);
                 if(x)
                 {
                     if(q[i])//如果i是本校学生(及有一张床)
                     {
                         add(j, i);//j也可以睡
                     }
                     if(q[j])//同理
                     {
                         add(i, j);
                     }
                 }
             }
         }
         for(register int i = ; i <= n; ++i)
         {
             if(q[i] && w[i])//如果是本校学生&&不留在学校(如果是外校的就不可能回家啊)
             {
                 ++cnt;//记录一下有多少个，以后算答案方便
                 continue;//不用管他有没有床
             }
             memset(vis, , sizeof(vis));
             if(!dfs(i))//如果有一个学生没有床
             {
                 printf("T_T\n");//哭了
                 break;
             }
             else
             {
                 ++ans;//记录有床的学生数量
             }
         }
         if(ans == n - cnt)//如果留学校的人都有床
         {
             printf("^_^\n");//笑了
         }
     }
     return ;
 }