【补充上一篇】网络流问题——最大带权闭合路径 2018 ACM-ICPC 中国大学生程序设计竞赛线上赛 F. Clever King

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/HZEUHJEJNJ/article/details/80092081

今天把吉林大学的模板也试了一下,它使用的是所谓的Dinic算法,比普通的BFS不知道快(高)到哪里去了
代码:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 400+5 
#define typec long long // type of cost
const typec inf = 0x3f3f3f3f; // max of cost
struct edge {
    int x, y, nxt;
    typec c;
} bf[5000];
int ne, head[N], cur[N], ps[N], dep[N];
void addedge(int x, int y, typec c) {
    // add an arc(x -> y, c); vertex: 0 ~ n-1;
    bf[ne].x = x;
    bf[ne].y = y;
    bf[ne].c = c;
    bf[ne].nxt = head[x];
    head[x] = ne++;
    bf[ne].x = y;
    bf[ne].y = x;
    bf[ne].c = 0;
    bf[ne].nxt = head[y];
    head[y] = ne++;
}
typec flow(int n, int s, int t) {
    typec tr, res = 0;
    int i, j, k, f, r, top;
    while (1) {
        memset(dep, -1, n * sizeof(int));
        for (f = dep[ps[0] = s] = 0, r = 1; f != r; )
            for (i = ps[f++], j = head[i]; j; j = bf[j].nxt) {
                if (bf[j].c && -1 == dep[k = bf[j].y]) {
                    dep[k] = dep[i] + 1;
                    ps[r++] = k;
                    if (k == t) {
                        f = r;
                        break;
                    }
                }
            }
        if (-1 == dep[t]) break;
        memcpy(cur, head, n * sizeof(int));
        for (i = s, top = 0; ; ) {
            if (i == t) {
                for (k = 0, tr = inf; k < top; ++k)
                    if (bf[ps[k]].c < tr)
                        tr = bf[ps[f = k]].c;
                for (k = 0; k < top; ++k)
                    bf[ps[k]].c -= tr, bf[ps[k]^1].c += tr;
                res += tr;
                i = bf[ps[top = f]].x;
            }
            for (j=cur[i]; cur[i]; j = cur[i] = bf[cur[i]].nxt)
                if (bf[j].c && dep[i]+1 == dep[bf[j].y]) break;
            if (cur[i]) {
                ps[top++] = cur[i];
                i = bf[cur[i]].y;
            } else {
                if (0 == top) break;
                dep[i] = -1;
                i = bf[ps[--top]].x;
            }
        }
    }
    return res;
}
int main() {
    int T;
    int n, m, a, b, t;
    long long sum;
    long long mincut;
    cin >> T;
    while (T--) {
        sum = 0;
        ne = 2;
        memset(head,0,sizeof(head));
        memset(bf,0,sizeof(bf));
        cin >> n >> m;
        for (int i = 1; i <= n; ++i) {
            cin >> t;
            addedge(0,i,t);
            sum += t;
        }
        for (int i = 1; i <= m; ++i) {
            cin >> t;
            addedge(i+n,n+m+1,t);
        }
        for (int i = 1; i <= n; ++i) {
            cin >> a >> b;
            for (int j = 0; j<a; ++j) {
                cin >> t;
                addedge(i,t+n,inf);
            }
            for (int j = 0; j<b; ++j) {
                cin >> t;
                addedge(i,t,inf);
            }
        }
        mincut = sum - flow(n+m+2,0,n+m+1);
        cout << mincut << endl;
    }
    return 0;
}

最后是一个比较
吉林大学模板:
这里写图片描述
浙江大学模板
这里写图片描述
刘汝佳书上模板:
这里写图片描述

猜你喜欢

转载自blog.csdn.net/HZEUHJEJNJ/article/details/80092081