BZOJ1433[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,但是显然自己还是可以睡自己的床),
认识的关系是相互的。
1 ≤ n ≤ 50,1 ≤ T ≤ 20

输出

对于每组数据,如果存在一个方案则输出“^_^”(不含引号)否则输出“T_T”(不含引号)。
(注意输出的都是半角字符,即三个符号的ASCII码分别为94,84,95)

样例输入

1
3
1 1 0
0 1 0
0 1 1
1 0 0
1 0 0

样例输出

^_^
 
 
  题意显然是求二分图最大匹配,重点是建模。源点连向住校的在校生和外校生表示这些人需要床,在校生的床连向汇点表示这些人有床,对于每个人连向认识的人的床(不要忘了连自己的床)。最后答案就是二分图最大匹配。
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int k;
int n,t;
int S,T;
int ans;
int tot=1;
int a[60];
int b[60];
int q[1000];
int d[1000];
int c[60][60];
int head[1000];
int to[100000];
int val[100000];
int next[100000];
int INF=2147483647;
void add(int x,int y,int z)
{
    tot++;
    next[tot]=head[x];
    head[x]=tot;
    to[tot]=y;
    val[tot]=z;
    tot++;
    next[tot]=head[y];
    head[y]=tot;
    to[tot]=x;
    val[tot]=0;
}
bool bfs(int S,int T)
{
    memset(d,-1,sizeof(d));
    memset(q,0,sizeof(q));
    int r=0;
    int l=0;
    q[r++]=S;
    d[S]=0;
    while(l<r)
    {
        int now=q[l];
        l++;
        for(int i=head[now];i;i=next[i])
        {
            if(d[to[i]]==-1&&val[i]!=0)
            {
                q[r++]=to[i];
                d[to[i]]=d[now]+1;
            }
        }
    }
    if(d[T]!=-1)
    {
        return true;
    }
    else
    {
        return false;
    }
}
int dfs(int x,int maxflow)
{
    if(x==T)
    {
        return maxflow;
    }
    int used=0;
    int nowflow;
    for(int i=head[x];i;i=next[i])
    {
        if(d[to[i]]==d[x]+1&&val[i]!=0)
        {
            nowflow=dfs(to[i],min(val[i],maxflow-used));
            val[i]-=nowflow;
            val[i^1]+=nowflow;
            used+=nowflow;
            if(nowflow==maxflow)
            {
                return nowflow;
            }
        }
    }
    if(used==0)
    {
        d[x]=-1;
    }
    return used;
}
void dinic()
{
    while(bfs(S,T)==true)
    {
        ans+=dfs(S,INF);
    }
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        ans=0;
        k=0;
        memset(head,0,sizeof(head));
        scanf("%d",&n);
        S=2*n+1;
        T=2*n+2;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&b[i]);
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&c[i][j]);
            }
        }
        for(int i=1;i<=n;i++)
        {
            if(a[i])
            {
                add(n+i,T,1);
                if(!b[i])
                {
                    add(S,i,1);
                    add(i,n+i,1);
                    k++;
                }
            }
            else
            {
                add(S,i,1);
                k++;
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(c[i][j])
                {
                    add(i,n+j,1);
                }
            }
        }
        dinic();
        if(ans==k)
        {
            printf("^_^\n");
        }
        else
        {
            printf("T_T\n");
        }
    }
}
posted @ 2018-07-12 20:43  The_Virtuoso  阅读(96)  评论(0编辑  收藏