-网络流-最小割- [洛谷 P2057][SHOI2007]善意的投票

题目描述

幼儿园里有n个小朋友打算通过投票来决定睡不睡午觉。对他们来说，这个问题并不是很重要，于是他们决定发扬谦让精神。虽然每个人都有自己的主见，但是为了照顾一下自己朋友的想法，他们也可以投和自己本来意愿相反的票。我们定义一次投票的冲突数为好朋友之间发生冲突的总数加上和所有和自己本来意愿发生冲突的人数。

我们的问题就是，每位小朋友应该怎样投票，才能使冲突数最小？

输入格式

文件的第一行只有两个整数n，m，保证有2≤n≤300，1≤m≤n(n-1)/2。其中n代表总人数，m代表好朋友的对数。文件第二行有n个整数，第i个整数代表第i个小朋友的意愿，当它为1时表示同意睡觉，当它为0时表示反对睡觉。接下来文件还有m行，每行有两个整数i，j。表示i，j是一对好朋友，我们保证任何两对i，j不会重复。

输出格式

只需要输出一个整数，即可能的最小冲突数。

输入样例

3 3
1 0 0
1 2
1 3
3 2

输出样例

1

说明

2≤n≤300，1≤m≤n(n-1)/2。

Solution

这道题这么建模：

把源点都连向同意睡觉的人，再把所有不同意睡觉的人都连向汇点，再在朋友之间连边。

于是这道题就成了最小割。

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
#define MAXN 1000002
#define INF 2000000000
int min(int a, int b) {
    if (a < b) return a;
    else return b;
}
struct Edge {
    int v, nx, w;
} e[MAXN];
std::queue <int> q;
int n, m, head[MAXN], ecnt = 1, x, y, z, r, k, dep[MAXN], cur[MAXN], cnt = 1, totp = 0, tot, c, p, g;
void add(int f, int t, int w) {
    e[++ecnt] = (Edge) {t, head[f], w};
    head[f] = ecnt;
    e[++ecnt] = (Edge) {f, head[t], 0};
    head[t] = ecnt;
}
bool bfs(int s, int t) {
    memset(dep, 0x7f, sizeof(dep));
    while (!q.empty()) q.pop();
    for (int i = 1; i <= n + 2; i++) {
        cur[i] = head[i];
    }
    dep[s] = 0;
    q.push(s);
    while (!q.empty()) {
        int v = q.front();
        q.pop();
        for (int i = head[v]; i; i = e[i].nx) {
            int to = e[i].v;
            if (dep[to] > INF && e[i].w) {
                dep[to] = dep[v] + 1;
                q.push(to);
            }
        }
    }
    if (dep[t] < INF) return 1;
    else return 0;
}
int dfs(int u, int t, int l) {
    if (!l || u == t) return l;
    int fl = 0, f;
    for (int i = cur[u]; i; i = e[i].nx) {
        cur[u] = i;
        int to = e[i].v;
        if (dep[to] == dep[u] + 1 && (f = dfs(to, t, min(l, e[i].w)))) {
            fl += f;
            l -= f;
            e[i ^ 1].w += f;
            e[i].w -= f;
            if (!l) break;
        }
    }
    return fl;
}
int Dinic(int s, int t) {
    int maxf = 0;
    while (bfs(s, t)) {
        maxf += dfs(s, t , INF);
    }
    return maxf;
}
int h(int i, int j) {
    return (i - 1) * m + j;
}
char a, b;
int main() {
    scanf("%d%d", &n, &m);
    r = n + 1;
    k = n + 2;
    for (int i = 1; i <= n; i++) {
        scanf("%d", &x);
        if (x == 1) {
            add(r, i, 1);
        }
        else {
            add(i, k, 1);
        }
    }
    for (int i = 1; i <= m; i++) {
        scanf("%d%d", &x, &y);
        add(x, y, 1);
        add(y, x, 1);
    }
    tot = Dinic(r, k);
    printf("%d\n", tot);
    return 0;
}