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

Материал из Олимпиадное программирование в УлГТУ
Перейти к навигации Перейти к поиску
Нет описания правки
Нет описания правки
 
(не показана 1 промежуточная версия этого же участника)
Строка 1: Строка 1:
  class Graph {
  class Graph {
     struct Edge {
     struct Edge {
         int a, b, capacity, flow;
         int a, b, capacity, flow = 0;
   
   
         Edge(int a, int b, int capacity) :
         Edge(int a, int b, int capacity) :
             a(a), b(b), capacity(capacity), flow(0) {}
             a(a), b(b), capacity(capacity) {}
   
   
         int other(int v) const {
         int other(int v) const {
Строка 14: Строка 14:
         }
         }
   
   
         void addFlowTo(int v, int f) {
         void addFlowTo(int v, int deltaFlow) {
             flow += (v == b ? f : -f);
             flow += (v == b ? deltaFlow : -deltaFlow);
         }
         }
     };
     };
   
   
     vector<Edge> edges;
     vector<Edge> edges;
     vector<vector<int>> g;
     vector<vector<int>> graph;
     vector<bool> visited;
     vector<bool> visited;
     vector<int> edgeTo;
     vector<int> edgeTo;
Строка 26: Строка 26:
     void dfs(int v) {
     void dfs(int v) {
         visited[v] = 1;
         visited[v] = 1;
         for (int e : g[v]) {
         for (int e : graph[v]) {
             int to = edges[e].other(v);
             int to = edges[e].other(v);
             if (!visited[to] && edges[e].capacityTo(to) > 0) {
             if (!visited[to] && edges[e].capacityTo(to)) {
                 edgeTo[to] = e;
                 edgeTo[to] = e;
                 dfs(to);
                 dfs(to);
Строка 35: Строка 35:
     }
     }
   
   
     bool hasPath(int from, int to) {
     bool hasPath(int start, int finish) {
         fill(visited.begin(), visited.end(), 0);
         visited.assign(visited.size(), 0);
         dfs(from);
         dfs(start);
         return visited[to];
         return visited[finish];
     }
     }
   
   
     int bottleneckCapacity(int from, int to) {
     int bottleneckCapacity(int start, int finish) {
         int bCapacity = 1e9;
         int bCapacity = 1e9;
         for (int v = to; v != from; v = edges[edgeTo[v]].other(v))
         for (int v = finish; v != start; v = edges[edgeTo[v]].other(v))
             bCapacity = min(bCapacity, edges[edgeTo[v]].capacityTo(v));
             bCapacity = min(bCapacity, edges[edgeTo[v]].capacityTo(v));
         return bCapacity;
         return bCapacity;
     }
     }
   
   
     void addFlow(int from, int to, int flow) {
     void addFlow(int start, int finish, int deltaFlow) {
         for (int v = to; v != from; v = edges[edgeTo[v]].other(v))
         for (int v = finish; v != start; v = edges[edgeTo[v]].other(v))
             edges[edgeTo[v]].addFlowTo(v, flow);
             edges[edgeTo[v]].addFlowTo(v, deltaFlow);
     }
     }
   
   
  public:
  public:
     Graph(int vertexCount) {
     Graph(int vertexCount) :
         g.resize(vertexCount);
         graph(vertexCount), visited(vertexCount), edgeTo(vertexCount) {}
        visited.resize(vertexCount);
       
        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));
         g[from].push_back(edges.size() - 1);
         graph[from].push_back(edges.size() - 1);
         g[to].push_back(edges.size() - 1);
         graph[to].push_back(edges.size() - 1);
     }
     }
   
   
     long long maxFlow(int from, int to) {
     long long maxFlow(int start, int finish) {
         long long flow = 0;
         long long flow = 0;
         while (hasPath(from, to)) {
         while (hasPath(start, finish)) {
             int deltaFlow = bottleneckCapacity(from, to);
             int deltaFlow = bottleneckCapacity(start, finish);
             addFlow(from, to, deltaFlow);
             addFlow(start, finish, deltaFlow);
             flow += deltaFlow;
             flow += deltaFlow;
         }
         }

Текущая версия от 13:23, 1 июля 2023

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

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

        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 deltaFlow) {
            flow += (v == b ? deltaFlow : -deltaFlow);
        }
    };

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

    void dfs(int v) {
        visited[v] = 1;
        for (int e : graph[v]) {
            int to = edges[e].other(v);
            if (!visited[to] && edges[e].capacityTo(to)) {
                edgeTo[to] = e;
                dfs(to);
            }
        }
    }

    bool hasPath(int start, int finish) {
        visited.assign(visited.size(), 0);
        dfs(start);
        return visited[finish];
    }

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

    void addFlow(int start, int finish, int deltaFlow) {
        for (int v = finish; v != start; v = edges[edgeTo[v]].other(v))
            edges[edgeTo[v]].addFlowTo(v, deltaFlow);
    }

public:
    Graph(int vertexCount) :
        graph(vertexCount), visited(vertexCount), edgeTo(vertexCount) {}
        
    void addEdge(int from, int to, int capacity) {
        edges.push_back(Edge(from, to, capacity));
        graph[from].push_back(edges.size() - 1);
        graph[to].push_back(edges.size() - 1);
    }

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

Ссылки

Теория:

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

Код:

Задачи:

Источник — https://acm.khpnets.info/w39/index.php?title=Алгоритм_Форда-Фалкерсона&oldid=2901