BZOJ-1433 假期的宿舍 最大流+基础建图

网络流练习ing。。

1433: [ZJOI2009]假期的宿舍
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 1748 Solved: 765
[Submit][Status][Discuss]

Description

Input

Output

Sample Input
1
3
1 1 0
0 1 0
0 1 1
1 0 0
1 0 0

**Sample Outpu**t
ˆ-ˆ

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

基础建图：
超级源连向所有在校生（有床位的人）；所有在校生且不回家的，与所有非在校生的与超级源相连；每个点分成两个i与i'，如果i和j认识连i与j'；边权全部为1
最大流即为答案。

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define M 1000010
struct data{
     int  to,next,v;
}edge[M];
int head[M],cnt=1;
int dis[M];
int q[M],h,t;
int c,n,ans,num;
int zx[60],know[60][60],hj[60];
 //zx【】是否在校，know【】【】是否认识，hj【】是否回家

void add(int u,int v,int w)
{
    cnt++;
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
    edge[cnt].v=w;
}

void make()//超级源为0，超级汇为2*n+1
{
    cnt=1;  memset(head,0,sizeof(head));
    num=0;
    for (int i=1; i<=n; i++)
        for (int j=1; j<=n; j++)
            if (know[i][j]==1)
                {
                    add(i,j+n,1);
                    add(j+n,i,0);
                }
    for (int i=1; i<=n; i++)
        {
            if (zx[i]==1)
                {
                    add(0,i,1);
                    add(i,0,0);
                }
            if ((zx[i]==1 && hj[i]==0) || (zx[i]==0))
                {
                    add(i+n,2*n+1,1);
                    add(2*n+1,i+n,0);
                    num++;//需要床位数
                }
        }
}

bool bfs()
{
    memset(dis,-1,sizeof(dis));
    q[1]=0; dis[0]=1;
    h=0; t=1;
    while (h<t)
        {
            int j=q[++h],i=head[j];
            while (i)
                {
                    if (edge[i].v>0 && dis[edge[i].to]<0)
                        {
                            dis[edge[i].to]=dis[j]+1;
                            q[++t]=edge[i].to;
                        }
                    i=edge[i].next;
                }
        }
    if (dis[2*n+1]>0)
        return true;
    else
        return false;   
}

int dfs(int loc,int low)
{
    int flow=0;
    if (loc==2*n+1) return low;
    int i=head[loc];
    while (i)
        {
            if (edge[i].v>0 && dis[edge[i].to]==dis[loc]+1 && (flow=dfs(edge[i].to,min(low,edge[i].v))))
                {
                    edge[i].v-=flow;
                    edge[i^1].v+=flow;
                    return flow;
                }
            i=edge[i].next;
        }
    return 0;
}

int main()
{
    scanf("%d",&c);
    while (c--)
        {
            scanf("%d",&n);
            memset(zx,0,sizeof(zx));
            memset(hj,0,sizeof(hj));
            memset(know,0,sizeof(know));
            for (int i=1; i<=n; i++)
                scanf("%d",&zx[i]);
            for (int i=1; i<=n; i++)
                scanf("%d",&hj[i]);
            for (int i=1; i<=n; i++)
                for (int j=1; j<=n; j++)
                    scanf("%d",&know[i][j]);
            for (int i=1; i<=n; i++)
                know[i][i]=1;//由题意自己和自己在读入时为不能睡，但实际可以，修改。
            make();
            ans=0;
            while (bfs())
                {
                    int now=0;
                    while ((now=dfs(0,0x7fffffff)))
                        ans+=now;
                }
            if  (num==ans)
                puts("^_^");
            else
                puts("T_T");        
        }
    return 0;
}