Uva-10480 Sabotage

題目鏈接:Sabotage

題目大意:求s與t點之間消除哪些邊使s與t不相連且所刪除的邊的權值和最小。

解題思路:這道題其實是求最小割的邊集,由於最小割等於最大流,所以跑一遍Dinic就可以得到最小權值和,但是如何求最小割的邊集呢?我們把最後的殘餘網絡找到,會發現那些最小權值且能使s與t聯通的邊權已經變爲了0,那麼我們就可以把與s點相連且相連邊的權不爲零的點集找到,再把這些點相連的且邊權爲零的邊找到,即爲所求答案。

代碼如下:

#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int inf  = 0x3f3f3f3f;
const int maxn = 100 + 15;
const int maxm = 1e6 + 15;
struct Edge {
    int to, cap, next;
};

int ecnt, st, ed, m, n;
Edge es[maxm];
int cur[maxn], dep[maxn], head[maxn], vis[maxn]; 

class Dinic {
public:
    void init(){
        memset(head, -1, sizeof head);
        ecnt = 0;
    }

    void add_edge(int u, int v, int cap) {
        es[ecnt].to = v;
        es[ecnt].next = head[u];
        es[ecnt].cap = cap;
        head[u] = ecnt++;
    }

    void add_double(int u, int v, int w1, int w2 = 0) {
        add_edge(u, v, w1);
        add_edge(v, u, w2);
    }

    bool BFS() {
        queue<int> q; q.push(st);
        memset(dep, 0x3f, sizeof dep); dep[st] = 0;
        while(q.size() && dep[ed] == inf) {
            int u = q.front(); q.pop();
            for(int i = head[u]; ~i; i = es[i].next) {
                Edge& e = es[i];
                if(e.cap > 0 && dep[e.to] == inf) {
                    dep[e.to] = dep[u] + 1;
                    q.push(e.to);
                }
            }
        }
        return dep[ed] < inf;
    }

    int DFS(int u, int maxflow) {
        if(u == ed) return maxflow;
        int res = 0;
        for(int i = cur[u]; ~i; i = es[i].next) {
            Edge& e = es[i];
            if(dep[e.to] == dep[u] + 1 && e.cap > 0) {
                int flow = DFS(e.to, min(maxflow, e.cap));  
                cur[u] = i; 
                res += flow; maxflow -= flow;
                es[i].cap -= flow; 
                es[i ^ 1].cap += flow;
                if(!maxflow) return res;
            }
        }       
        dep[u] = inf;
        return res;
    }

    ll MaxFlow(){
        ll ans = 0; 
        while(BFS()) {   
            for(int i = st; i <= ed; i++) cur[i] = head[i];
            ans += DFS(st, inf);
        }
        return ans;
    }
};

inline int read() {
    char c = getchar();
    while(!isdigit(c)) c = getchar();
    int x = 0;
    while(isdigit(c)) {
        x = x * 10 + c - '0';
        c = getchar();
    }
    return x;
}

int main(){
#ifdef NEKO
    freopen("Nya.txt", "r", stdin);
#endif
    Dinic dic;
    while(scanf("%d%d", &n, &m) != EOF && n + m) {
        st = 0, ed = n + 1; dic.init();
        dic.add_double(st, 1, inf);
        dic.add_double(2, ed, inf);
        for(int i = 1; i <= m; i++) {
            int u, v, w; u = read();
             v = read(); w = read();
            dic.add_double(u, v, w, w);
        }
        dic.MaxFlow();
        set<int> sp; vector<P> ans;
        queue<int> que; que.push(st);
        while(que.size()) {
            int u = que.front(); que.pop();
            sp.insert(u);
            for(int i = head[u]; ~i; i = es[i].next) {
                int v = es[i].to;
                if(es[i].cap && !sp.count(v)) {
                    sp.insert(v);   
                    que.push(v);
                }
            }
        }
        que.push(st); memset(vis, 0, sizeof vis);
        while(que.size()) {
            int u = que.front(); que.pop(); vis[u] = 1;
            for(int i = head[u]; ~i; i = es[i].next) {
                int v = es[i].to;
                if(!es[i].cap && !sp.count(v))
                    ans.push_back(P(u, v));
                else if(!vis[v]) {  
                    vis[v] = 1;
                    que.push(v);
                }
            }
        }       
        sort(ans.begin(), ans.end());
        ans.resize(unique(ans.begin(), ans.end()) - ans.begin());
        for(int i = 0; i < ans.size(); i++)
            printf("%d %d\n", ans[i].first, ans[i].second); 
        puts("");

    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章