[ZJOI2009]假期的宿舍(luogu P2055)

原题链接:https://www.luogu.org/problem/show?pid=2055

其实本题除了建边之外几乎与二分图最大匹配模板题没什么区别。

二分图的一边的人,一边为床。。。

如果本校学生且回家,那么就没他什么事情了,他并不需要床铺。如果是本校学生但不回家,那么就可以向自己的床连一条边,因为下面的邻接矩阵中并没有给出这种方案,但它是存在的。外校学生的话,就要向自己认识的人的床连一条边。

之后跑二分图最大匹配即可,检查匹配数是否能满足需求。

本题有多组数据,一定记得把head,tot,cnt这些东西清空。不然就会像我在后面的几组数据中WA到怀疑人生。

 

#include<cstdio>
#include<cstring>
void read(int &y)
{
    y=0;char x=getchar();
    while(x<'0'||x>'9') x=getchar();
    while(x>='0'&&x<='9')
    {
        y=y*10+x-'0';
        x=getchar();
    }
}
struct edge
{
    int u,v;
}e[10005];
int t,n,tot,sum,x,cnt;
int d[55],z[55];
int vis[105],head[105],match[105];
void add(int u,int v)
{
    e[++cnt].u=head[u];
    e[cnt].v=v;
    head[u]=cnt;
}
int dfs(int x)
{
    for(int i=head[x];i;i=e[i].u)
    {
        int t=e[i].v;
        if(vis[t]==1) continue;
        vis[t]=1;
        if(match[t]==0||dfs(match[t]))
        {
            match[t]=x;
            return 1;
        } 
    }
    return 0;
}
int main()
{
    read(t);
    while(t--)
    {
        read(n);
        memset(head,0,sizeof(head));cnt=0;tot=0;//WA WA WA
        for(int i=1;i<=n;i++) read(d[i]);
        for(int i=1;i<=n;i++)
        {
            read(z[i]);
            if(z[i]==0&&d[i]==1) add(i,i);
            if(d[i]==0||(z[i]==0&&d[i]==1)) tot++;
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                read(x);
                if(x==1&&d[j]==1) add(i,j);
            }
        }
        memset(match,0,sizeof(match));
        sum=0;
        for(int i=1;i<=n;++i)
        {
            if((d[i]==1&&z[i]==0)||d[i]==0)
            {
                memset(vis,0,sizeof(vis));
                if(dfs(i)) sum++;
            }
        }
        if(sum>=tot) printf("^_^\n");
        else printf("T_T\n");
    }
    return 0;
}

 

posted @ 2017-10-24 09:59  Excim  阅读(139)  评论(0编辑  收藏  举报