AC日记——[ZJOI2009]假期的宿舍 cogs 1333

1333. [ZJOI2009] 假期的宿舍

★★☆   输入文件：zjoi09holiday.in   输出文件：zjoi09holiday.out   简单对比
时间限制：1 s   内存限制：256 MB

思路：

　　网络流裸题；

来，上代码：

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

#define maxn 105
#define INF 0x7fffffff

using namespace std;

struct EdgeType {
    int v,f,e;
};
struct EdgeType edge[maxn*maxn*];

int n,home[maxn],s,t=maxn-,deep[maxn],head[maxn],cnt,T;

char Cget;

inline void in(int &now)
{
    now=,Cget=getchar();
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}

inline void edge_add(int u,int v,int f)
{
    edge[++cnt].v=v,edge[cnt].f=f,edge[cnt].e=head[u],head[u]=cnt;
    edge[++cnt].v=u,edge[cnt].f=,edge[cnt].e=head[v],head[v]=cnt;
}

bool bfs()
{
    queue<int>que;
    for(int i=s;i<=t;i++) deep[i]=-;
    que.push(s),deep[s]=;
    while(!que.empty())
    {
        int now=que.front();que.pop();
        for(int i=head[now];i;i=edge[i].e)
        {
            if(deep[edge[i].v]<&&edge[i].f>)
            {
                deep[edge[i].v]=deep[now]+;
                if(edge[i].v==t) return true;
                que.push(edge[i].v);
            }
        }
    }
    return false;
}

int flowing(int now,int flow)
{
    if(now==t||flow<=) return flow;
    int oldflow=;
    for(int i=head[now];i;i=edge[i].e)
    {
        if(deep[edge[i].v]!=deep[now]+||edge[i].f<=) continue;
        int pos=flowing(edge[i].v,min(flow,edge[i].f));
        if(pos>)
        {
            flow-=pos;
            oldflow+=pos;
            edge[i].f-=pos;
            edge[i^].f+=pos;
            if(flow==) return oldflow;
        }
    }
    if(oldflow==) deep[now]=-;
    return oldflow;
}

int main()
{
    freopen("zjoi09holiday.in","r",stdin);
    freopen("zjoi09holiday.out","w",stdout);
    in(T);
    while(T--)
    {
        in(n);int lit=n;cnt=;
        for(int i=s;i<=t;i++) head[i]=;
        for(int i=;i<=n;i++)
        {
            in(home[i]);
            if(home[i]) edge_add(i+n,t,);
        }
        int pos;
        for(int i=;i<=n;i++)
        {
            in(pos);
            if(home[i])
            {
                if(!pos) edge_add(s,i,);
                else lit--;
            }
            else edge_add(s,i,);
        }
        for(int i=;i<=n;i++)
        {
            if(home[i]) edge_add(i,i+n,);
            for(int j=;j<=n;j++)
            {
                in(pos);
                if(pos) edge_add(i,j+n,);
            }
        }
        pos=;
        while(bfs()) pos+=flowing(s,INF);
        if(pos>=lit) printf("^_^\n");
        else printf("T_T\n");
    }
    fclose(stdin);fclose(stdout);
    return ;
}