最小費用最大流 模板

  1. 因爲含有負權邊,所以使用spfa進行增廣。
  2. 指定流量的最小費用流可以初始化一個f,然後每次一直循環到f<= 0爲止。
  3. 函數返回的是最大流,當然經過少量修改可以返回最小費用,利用最小流量乘以相應的費用即可。
  4. prevv記錄父節點,preve記錄當前節點對應父節點的第幾條邊。
const int INF = 0x3fffffff;
struct Edge {
    int to, cap, cost, rev;
    Edge(int too, int capp, int costt, int revv) : to(too), cap(capp), cost(costt), rev(revv) {}
};

vector<vector<Edge> > G;
int st, en;
int dist[30];
int prevv[30], preve[30];

void add_edge(int from, int to, int cap, int cost) {
    G[from].push_back(Edge(to, cap, cost, G[to].size()));
    G[to].push_back(Edge(from, 0, -cost, G[from].size()-1));
}

int min_max_flow(int n) {//the quantity of points.
    int ans = 0;//return the greatest flow.
    while(true) {//spfa
        fill(dist, dist+n, INF);
        dist[st] = 0;
        queue<int> q;  q.push(st);

        while(!q.empty()) {
            int u = q.front();  q.pop();
            for(int i = 0; i < G[u].size(); i++) {
                int to = G[u][i].to;
                if(G[u][i].cap > 0 && dist[u] + G[u][i].cost < dist[to]) {
                    dist[to] = dist[u] + G[u][i].cost;
                    prevv[to] = u;  preve[to] = i;
                    q.push(to);
                }
            }
        }
        if(dist[en] == INF)
            break;
        else {
            int f = INF;
            for(int pr = en; pr != st; pr = prevv[pr])
                f = min(f, G[prevv[pr]][preve[pr]].cap);
            ans += f;
            for(int pr = en; pr != st; pr = prevv[pr]) {
                Edge &e = G[prevv[pr]][preve[pr]];
                e.cap -= f;
                G[pr][e.rev].cap += f;
            }
        }
    }
    return ans;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章