bzoj1497 最大獲利(最大權閉合子圖)

wxyww發表於2019-02-08

題目連結

思路

對於每個中轉站向(T)連一條權值為建這個中轉站代價的邊。割掉這條邊表示會建這個中轉站。
對於每個人向他的兩個中轉站連一條權值為(INF)的邊。然後從(S)向這個人連一條權值為這個人的收益的邊，割掉這條邊表示不要這個收益。
這就是最大權閉合子圖的模型。
最後的答案=全部的收益-割掉的收益-建中轉站的代價=全部收益-最小割

程式碼

#include<cstring>
#include<algorithm>
#include<queue>
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<bitset>
using namespace std;
typedef long long ll;
const int N = 100010,M = 1000010,INF = 1e9;
ll read() {
    ll x=0,f=1;char c=getchar();
    while(c<`0`||c>`9`) {
        if(c==`-`) f=-1;
        c=getchar();
    }
    while(c>=`0`&&c<=`9`) {
        x=x*10+c-`0`;
        c=getchar();
    }
    return x*f;
}
struct node {
    int v,nxt,w;
}e[M << 1];
int head[N],ejs = 1;
void add(int u,int v,int w) {
    e[++ejs].v = v;e[ejs].nxt = head[u];head[u] = ejs;e[ejs].w = w;
    e[++ejs].v = u;e[ejs].nxt = head[v];head[v] = ejs;e[ejs].w = 0;
}
int S,T;
int n,m,cur[N];
queue<int>q;
int dep[N];
int bfs() {
    while(!q.empty()) q.pop();
    memset(dep,0,sizeof(dep));
    dep[S] = 1;q.push(S);
    while(!q.empty()) {
        int u = q.front();q.pop();
        for(int i = head[u];i;i = e[i].nxt) { 
            int v = e[i].v;
            if(!dep[v] && e[i].w) {
                dep[v] = dep[u] + 1;q.push(v);
                if(v == T) return 1;
            }
        }
    }
    return 0;
}
int dfs(int u,int now) {
    if(u == T) return now;
    int ret = 0;
    for(int &i = cur[u];i;i = e[i].nxt) {
        int v = e[i].v;
        if(dep[v] == dep[u] + 1 && e[i].w) {
            int k = dfs(v,min(now - ret,e[i].w));
            ret += k;
            e[i].w -= k;
            e[i ^ 1].w += k;
            if(ret == now) return ret;
        }
    }
    return ret;
}
int dinic() {
    int ans = 0;
    while(bfs()) {
        for(int i = 1;i <= T;++i) cur[i] = head[i];
        ans += dfs(S,INF);
    }
    return ans;
}
int main() {
    n = read(),m = read();
    T = n + m + 2,S = T - 1;
    int tot = 0;
    for(int i = 1;i <= n;++i) {
        int w = read();
        add(i,T,w);
    }
    for(int i = 1;i <= m;++i) {
        int x = read(),y = read(),w = read();
        add(i + n,x,INF);add(i + n,y,INF);
        add(S,i + n,w);
        tot += w;
    }
    cout<<tot - dinic();
    return 0;
}

相關文章