bzoj1433: [ZJOI2009]假期的宿舍

1433: [ZJOI2009]假期的宿舍

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 2315  Solved: 981
[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 Output

ˆ ˆ

HINT

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

难度在题意？

#include<bits/stdc++.h>
#define inf 2147483647
#define N 1280
#define rep(i,l,r) for(int i=l;i<=r;i++)
using namespace std;
int T,head[N],cnt,ans,ccz,n,tot=1,mp[N][N],f[N],g[N],dis[N];
struct node{
    int to,next,w;
}e[N*N<<1];
inline bool bfs(){
     for(int i=0;i<=T;i++) dis[i]=-1; queue<int>q; q.push(0); dis[0]=0;
     while(!q.empty()) {
          int x=q.front(); q.pop();
          for(int k=head[x];k;k=e[k].next) 
             if(dis[e[k].to]<0 && e[k].w>0) {
                   dis[e[k].to]=dis[x]+1; q.push(e[k].to);
             }
     }
     if(dis[T]>0) return 1;else return 0;
}
int find(int x,int low){
     if(x==T) return low;
     int delta=low,now;
     for(int k=head[x];k;k=e[k].next) 
       if(e[k].w>0 && dis[e[k].to]==dis[x]+1){ 
           now=find(e[k].to,min(e[k].w,delta));
           e[k].w-=now; e[k^1].w+=now;   delta-=now;
           if(!delta) return low;
        } 
     dis[x]=-1;
     return low-delta;
}
inline void ins(int u,int v,int w) {
     e[++tot].to=v; e[tot].next=head[u]; head[u]=tot; e[tot].w=w;
}
inline void insert(int u,int v,int w) {
     ins(u,v,w); ins(v,u,0);
}
int main () { 
     scanf("%d",&ccz);
     while(ccz--) {
          scanf("%d",&n); memset(head,0,sizeof(head)); tot=1; cnt=ans=0; T=n+n+1;
          rep(i,1,n) scanf("%d",&f[i]);
          rep(i,1,n) scanf("%d",&g[i]);
          rep(i,1,n) rep(j,1,n) scanf("%d",&mp[i][j]),mp[i][j]+=(i==j);
          rep(i,1,n) if(!f[i] || (f[i]&&!g[i])) rep(j,1,n) if(mp[i][j]&&f[j]) insert(i,j+n,1);
          rep(i,1,n) if(!f[i] || (f[i]&&!g[i])) ++cnt,insert(0,i,1);
           rep(i,1,n) insert(i+n,T,1);
           while(bfs()) ans+=find(0,inf);
           if(ans>=cnt) puts("^_^");else puts("T_T");
       }
}