洛谷OJ:P2055 [ZJOI2009]假期的宿舍(最大流)

洛谷OJ:P2055 [ZJOI2009]假期的宿舍(最大流)_第1张图片

分析:一道求二分图最大匹配的题,就是建模比较绕其他没什么,首先我们要清楚人和床是需要分开的,所有需要2*n个结点,我们把前1~n结点当成人,n+1~2*n结点当成床,那么问题就简单了,我们只需要将床与汇点连接,需要床的人与源点连接,再将人与自己的床以及认识的人的床连接即可,最后求出的最大流如果与需要床的人数相同的话就可以使每个人都有床睡

#include 
#include 
#include 
#include 
#define init(a,b) memset(a, b, sizeof(a))
#define Min(a,b) a que;

void init_network_flows() {
    tot = 0;
    cnt = -1;
    init(es, -1);
    init(head, -1);
}

void add(int f, int t, int c) {
    es[++cnt].next = head[f];
    es[cnt].t = t;
    es[cnt].c = c;
    head[f] = cnt;
}

void add_edge(int f, int t, int c) {
    add(f, t, c); //添加一条正向边
    add(t, f, 0); //添加一条反向边
}

bool bfs() {
    int v;
    init(depth, 0);
    while(!que.empty()) que.pop();
    que.push(s); depth[s] = 1;
    while(!que.empty()) {
        v = que.front(); que.pop();
        for(int i = head[v]; i != -1; i = es[i].next) {
            if(es[i].c > 0 && depth[es[i].t] == 0) {
                depth[es[i].t] = depth[v] + 1;
                que.push(es[i].t);
                if(es[i].t == t) return true;
            }
        }
    }
    return depth[t] > 0;
}

int dfs(int u, int dis) {
    int d;
    if(u == t) return dis;
    for(int i = head[u]; i != -1; i = es[i].next) {
        if(depth[es[i].t] == depth[u] + 1 && es[i].c != 0) {
            d = dfs(es[i].t, Min(dis, es[i].c));
            if(d > 0) {
                es[i].c -= d; //正向边操作
                es[i^1].c += d; //反向边操作
                return d; //向上传递
            }
        }
    }
    return 0; //没有增广路
}

int Dinic() {
    int rec = 0, d;
    while(bfs()) {
        while(d = dfs(s, inf))
            rec += d;
    }
    return rec;
}

void input() {
    int temp;
    scanf("%d", &n);
    s = 0; t = n*2+1;
    for(int i = 1; i <= n; i++) {
        scanf("%d", &is_student[i]);
        if(is_student[i]) add_edge(n+i, t, 1); //床与汇点连接
        else { add_edge(s, i, 1); tot++; } //如果不是学生原点与床连边
    }
    for(int i = 1; i <= n; i++) {
        scanf("%d", &temp);
        if(is_student[i] && !temp) {
            add_edge(s, i, 1);
            add_edge(i, i+n, 1); //
            tot++;
        }
    }
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            scanf("%d", &temp);
            if(temp) { //如果认识,那么连一条边
                add_edge(i, j+n, 1);
                //add_edge(i, j, 1) ?
            }
        }
    }
}

void solve() {
    int max_flow = Dinic();
    //printf("max_flow = %d, tot = %d\n", max_flow, tot);
    if(max_flow == tot) printf("^_^\n");
    else printf("T_T\n");
}

int main() {
    int t;
    scanf("%d", &t);
    while(t--) {
        init_network_flows();
        input();
        solve();
    }
    return 0;
}

你可能感兴趣的:(图论学习,洛谷OJ)