洛谷 P2055 [ZJOI2009]假期的宿舍

题目描述

学校放假了 · · · · · · 有些同学回家了，而有些同学则有以前的好朋友来探访，那么住宿就是一个问题。比如 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：

ˆ_ˆ

说明

对于 30% 的数据满足 1 ≤ n ≤ 12。

对于 100% 的数据满足 1 ≤ n ≤ 50,1 ≤ T ≤ 20。

多组数据！！！

---

【分析】
貌似有两种解法：这里介绍二分图匹配

所以用匈牙利KM算法

但重要的是构建二分图的问题 各位犇的构图都很厉害 我只是按我的想法构建的一个：

二分图 一个为学生 一个为床铺

当学生是在校生时 若他回家 则学生不在 床在 所以 床的一部分加入此点

若不回家 学生在 床在 两部分都加入此点

当学生不是在校生 则没床 有人 学生的部分加入此点

当接收认识的学生后 再把在校学生的map[i][i] 赋值为1 即 可以自己睡自己的床

然后KM进去

最后判断 若二分图中学生多于床 一定不成立 直接“T_T”

---

【代码】

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<algorithm>
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=55;
int Q,n;
int link[mxn],bed[mxn],stu[mxn];
int vis[mxn],sco[mxn],back[mxn],know[mxn][mxn];
inline bool dfs(int a)   //匈牙利算法 
{
    int i;
    fo(i,1,n)
      if(know[a][i] && !vis[i] && bed[i])
      {
            vis[i]=1;
            if(!link[i] || dfs(link[i]))
            {
                  link[i]=a;
                  return 1;
          }
      }
    return 0;
}
int main()
{
    int i,j;
    scanf("%d",&Q);
    while(Q--)
    {
        int ans=0,sum=0;
        memset(know,0,sizeof know);
        memset(bed,0,sizeof bed);
        memset(stu,0,sizeof stu);
        memset(link,0,sizeof link);
        scanf("%d",&n);
        fo(i,1,n) scanf("%d",&sco[i]);
        fo(i,1,n) scanf("%d",&back[i]);
        fo(i,1,n) fo(j,1,n) scanf("%d",&know[i][j]);
        fo(i,1,n)
        {
            if(sco[i] && back[i]) bed[i]=1;  //回家的
            if(sco[i] && !back[i]) bed[i]=stu[i]=1,sum++;
            if(!sco[i]) stu[i]=1,sum++;
        }
        fo(i,1,n) if(sco[i]) know[i][i]=1;  //自己睡自己床 
        fo(i,1,n)  //拜访的匹配空床位
          if(stu[i])
          {
              memset(vis,0,sizeof vis);
              if(dfs(i)) ans++;
          }
        if(ans==sum) printf("^_^\n");
        else printf("T_T\n");
    }
    return 0;
}