HDU - 3549 - Flow Problem (最大流模板题--Dinic算法实现)

题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=3549

代码:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
using namespace std;
const int maxn=2e3+10;
const int maxm=5e3+10;
struct edge{
    int to,next,cap,rev;
}e[maxm];
int n,m,cnt=0,head[maxn],dis[maxn],iter[maxn];
queue<int> q;
void ins(int x,int y,int z) {
    e[++cnt].to = y;
    e[cnt].next = head[x];
    head[x] = cnt;
    e[cnt].cap = z;
    e[cnt].rev = cnt + 1;
    e[++cnt].to = x;
    e[cnt].next = head[y];
    head[y] = cnt;
    e[cnt].cap = 0;
    e[cnt].rev = cnt - 1;
}
void bfs(int s) {
    memset(dis, -1, sizeof(dis));
    while (!q.empty())q.pop();
    dis[s] = 0;
    q.push(s);
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        for (int i = head[u]; i; i = e[i].next) {
            if (e[i].cap && dis[e[i].to] < 0) {
                q.push(e[i].to);
                dis[e[i].to] = dis[u] + 1;
            }
        }
    }
}
int dfs(int v,int t,int f) {
    if (v == t)return f;
    for (int i = (iter[v] ? iter[v] : head[v]); i; i = e[i].next, iter[v] = i) {
        if (e[i].cap && dis[v] < dis[e[i].to]) {
            int d = dfs(e[i].to, t, min(f, e[i].cap));
            if (d > 0) {
                e[i].cap -= d;
                e[e[i].rev].cap += d;
                return d;
            }
        }
    }
    return 0;
}
int max_flow(int s,int t) {
    int flow = 0;
    for (;;) {
        bfs(s);
        if (dis[t] < 0)return flow;
        memset(iter, 0, sizeof(iter));
        int f;
        while ((f = dfs(s, t, 1e9)) > 0)
            flow += f;
    }
}
int main() {
    int _;
    scanf("%d", &_);
    for (int sce = 1; sce <= _; ++sce) {
        memset(head, 0, sizeof(head));
        memset(e, 0, sizeof(e));
        cnt = 0;
        scanf("%d%d", &n, &m);
        for (int i = 1; i <= m; ++i) {
            int x, y, z;
            scanf("%d%d%d", &x, &y, &z);
            ins(x, y, z);
        }
        printf("Case %d: %d\n", sce, max_flow(1, n));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/moon_sky1999/article/details/81348989