Алгоритм Эдмондса-Карпа: различия между версиями

Материал из Олимпиадное программирование в УлГТУ
Перейти к навигации Перейти к поиску
Нет описания правки
Строка 1: Строка 1:
  #include <stdio.h>
  class Graph {
  #include <algorithm>
    struct Edge {
  #include <vector>
        int a, b, capacity, flow;
#include <queue>
   
using namespace std;
        Edge(int a, int b, int capacity) :
            a(a), b(b), capacity(capacity), flow(0) {}
   
        int other(int v) const {
            return v == a ? b : a;
        }
   
   
class Edge {
         int capacityTo(int v) const {
    int a, b, capacity, flow;
            return v == b ? capacity - flow : flow;
public:
        }
    Edge(int a, int b, int capacity) : a(a), b(b), capacity(capacity), flow(0) {}
    int other(int v) const {
        void addFlowTo(int v, int f) {
         return v == a ? b : a;
            flow += (v == b ? f : -f);
    }
        }
    int capacityTo(int v) const {
    };
        return v == b ? capacity - flow : flow;
    }
    void addFlowTo(int v, int f) {
        flow += (v == b ? f : -f);
    }
};
   
   
class Graph {
     vector<Edge> edges;
     vector<Edge> edges;
     vector<vector<int>> g;
     vector<vector<int>> g;
     vector<bool> used;
     vector<bool> visited;
     vector<int> edgeTo;
     vector<int> edgeTo;
     void bfs(int v) {
     void bfs(int v) {
         queue<int> q;
         queue<int> q;
         used[v] = 1;
         visited[v] = 1;
         q.push(v);
         q.push(v);
         while (!q.empty()) {
         while (!q.empty()) {
Строка 34: Строка 33:
             for (int e : g[v]) {
             for (int e : g[v]) {
                 int to = edges[e].other(v);
                 int to = edges[e].other(v);
                 if (!used[to] && edges[e].capacityTo(to)) {
                 if (!visited[to] && edges[e].capacityTo(to)) {
                     edgeTo[to] = e;
                     edgeTo[to] = e;
                     used[to] = 1;
                     visited[to] = 1;
                     q.push(to);
                     q.push(to);
                 }
                 }
Строка 42: Строка 41:
         }
         }
     }
     }
     bool hasPath(int from, int to) {
     bool hasPath(int from, int to) {
         fill(used.begin(), used.end(), 0);
         fill(visited.begin(), visited.end(), 0);
         bfs(from);
         bfs(from);
         return used[to];
         return visited[to];
     }
     }
     int bottleneckCapacity(int from, int to) {
     int bottleneckCapacity(int from, int to) {
         int bCapacity = 1e9;
         int bCapacity = 1e9;
Строка 53: Строка 54:
         return bCapacity;
         return bCapacity;
     }
     }
     void addFlow(int from, int to, int flow) {
     void addFlow(int from, int to, int flow) {
         for (int v = to; v != from; v = edges[edgeTo[v]].other(v))
         for (int v = to; v != from; v = edges[edgeTo[v]].other(v))
             edges[edgeTo[v]].addFlowTo(v, flow);
             edges[edgeTo[v]].addFlowTo(v, flow);
     }
     }
  public:
  public:
     Graph(int verticesCount) {
     Graph(int vertexCount) {
         g.resize(verticesCount);
         g.resize(vertexCount);
         used.resize(verticesCount);
         visited.resize(vertexCount);
         edgeTo.resize(verticesCount);
         edgeTo.resize(vertexCount);
     }
     }
     void addEdge(int from, int to, int capacity) {
     void addEdge(int from, int to, int capacity) {
         edges.push_back(Edge(from, to, capacity));
         edges.push_back(Edge(from, to, capacity));
Строка 68: Строка 72:
         g[to].push_back(edges.size() - 1);
         g[to].push_back(edges.size() - 1);
     }
     }
     long long maxFlow(int from, int to) {
     long long maxFlow(int from, int to) {
         long long flow = 0;
         long long flow = 0;
Строка 78: Строка 83:
     }
     }
  };
  };
int main() {
    int n, m;
    scanf("%d%d", &n, &m);
    Graph g(n);
    int a, b, c;
    for (int i = 0; i < m; i++) {
        scanf("%d%d%d", &a, &b, &c);
        g.addEdge(a - 1, b - 1, c);
    }
    printf("%lld", g.maxFlow(0, n - 1));
}


== Ссылки ==
== Ссылки ==

Версия от 02:48, 26 декабря 2021

class Graph {
    struct Edge {
        int a, b, capacity, flow;

        Edge(int a, int b, int capacity) :
            a(a), b(b), capacity(capacity), flow(0) {}

        int other(int v) const {
            return v == a ? b : a;
        }

        int capacityTo(int v) const {
            return v == b ? capacity - flow : flow;
        }

        void addFlowTo(int v, int f) {
            flow += (v == b ? f : -f);
        }
    };

    vector<Edge> edges;
    vector<vector<int>> g;
    vector<bool> visited;
    vector<int> edgeTo;

    void bfs(int v) {
        queue<int> q;
        visited[v] = 1;
        q.push(v);
        while (!q.empty()) {
            v = q.front();
            q.pop();
            for (int e : g[v]) {
                int to = edges[e].other(v);
                if (!visited[to] && edges[e].capacityTo(to)) {
                    edgeTo[to] = e;
                    visited[to] = 1;
                    q.push(to);
                }
            }
        }
    }

    bool hasPath(int from, int to) {
        fill(visited.begin(), visited.end(), 0);
        bfs(from);
        return visited[to];
    }

    int bottleneckCapacity(int from, int to) {
        int bCapacity = 1e9;
        for (int v = to; v != from; v = edges[edgeTo[v]].other(v))
            bCapacity = min(bCapacity, edges[edgeTo[v]].capacityTo(v));
        return bCapacity;
    }

    void addFlow(int from, int to, int flow) {
        for (int v = to; v != from; v = edges[edgeTo[v]].other(v))
            edges[edgeTo[v]].addFlowTo(v, flow);
    }

public:
    Graph(int vertexCount) {
        g.resize(vertexCount);
        visited.resize(vertexCount);
        edgeTo.resize(vertexCount);
    }

    void addEdge(int from, int to, int capacity) {
        edges.push_back(Edge(from, to, capacity));
        g[from].push_back(edges.size() - 1);
        g[to].push_back(edges.size() - 1);
    }

    long long maxFlow(int from, int to) {
        long long flow = 0;
        while (hasPath(from, to)) {
            int deltaFlow = bottleneckCapacity(from, to);
            addFlow(from, to, deltaFlow);
            flow += deltaFlow;
        }
        return flow;
    }
};

Ссылки

Теория:

Демонстрация:

Код:

Задачи: