BZOJ1433 ZJOI2009 假期的宿舍 网络流

题意:有三种人:1、该校回家生  2、该校不回家生  3、非该校住宿生。给出人的认识关系,假若a认识b,则b(a)可以睡在a(b)的床上,只有该校的学生有床,回家生不占床位,求是否能让所有留宿的人都有床睡

题解:建二分图,左边是宿舍右边是学生。从S向需要宿舍的学生连边,从能提供宿舍的位置向T连边,如果某个学生能住某个宿舍则从学生向宿舍连边,跑最大流看是否满流即可。

#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <climits>
#include <iostream>
#include <algorithm>
using namespace std;

const int U=100+1;
const int MAXN=2000+2;
const int MAXM=6000000+2;
struct EDGE{
    int u,c;
    EDGE(){}
    EDGE(int _u,int _c):u(_u),c(_c){}
}e[MAXM];
int N,T,sum,d[MAXN],cnt,cur[MAXN];
bool t[MAXN];
queue<int> q;
vector<int> tab[MAXN];

void Insert(int u,int v,int c){
    tab[u].push_back(cnt),e[cnt++]=EDGE(v,c);
    tab[v].push_back(cnt),e[cnt++]=EDGE(u,0);
}

bool BFS(int s,int t){
    memset(d,-1,sizeof(d));
    d[s]=0,q.push(s);

    int x;
    while(!q.empty()){
        x=q.front(),q.pop();
        for(int i=0;i<tab[x].size();i++)
            if(d[e[tab[x][i]].u]==-1 && e[tab[x][i]].c)
                d[e[tab[x][i]].u]=d[x]+1,q.push(e[tab[x][i]].u);
    }
    return d[t]>0;
}

int DFS(int x,int f,int t){
    if(x==t) return f;

    int flow,used=0;
    for(int i=cur[x];i<tab[x].size();i++)
        if(e[tab[x][i]].c && d[e[tab[x][i]].u]==d[x]+1){
            flow=DFS(e[tab[x][i]].u,min(f-used,e[tab[x][i]].c),t);
            e[tab[x][i]].c-=flow,e[tab[x][i]^1].c+=flow,used+=flow;
            if(e[tab[x][i]].c) cur[x]=i;
            if(f==used) return f;
        }

    if(!used) d[x]=-1;
    return used;
}

int Dinic(int s,int t){
    int ret=0;
    while(BFS(s,t)){
        memset(cur,0,sizeof(cur));
        ret+=DFS(s,INT_MAX,t);
    }
    return ret;
}

int main(){
    cin >> T;
    while(T--){
        cin >> N;
        for(int i=1;i<=N;i++){
            scanf("%d",t+i);
            if(t[i]) Insert(N+i,U,1);
        }
        for(int i=1,w;i<=N;i++){
            scanf("%d",&w);
            if((!w && t[i]) || !t[i]) Insert(0,i,1),sum++;
        }
        for(int i=1,w;i<=N;i++)
            for(int j=1;j<=N;j++){
                scanf("%d",&w);
                if(w || i==j) Insert(i,j+N,1);
            }

        if(Dinic(0,U)==sum) puts("^_^");
        else puts("T_T");

        cnt=sum=0;
        for(int i=0;i<=U;i++) tab[i].clear();
    }

    return 0;
}
View Code

 

posted @ 2017-02-28 22:37  WDZRMPCBIT  阅读(41)  评论(0编辑  收藏