manasimu 0.0.36 → 0.0.38

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 82eaccd10f2cf79a10c97f388a08350673a287f547009ed7b033d76a4554290e
-  data.tar.gz: a64526ba25f0253c76a042bee15d4bb740710e4709cfe0702fdf8b970e6b7771
+  metadata.gz: d6ae84013078ea01136343e8a484fdd74e91b52030dbde9d6fd15fe1af8f511b
+  data.tar.gz: 4ca268428f78118faa37f018cd993870eb679f2ffffd269137fe256edb021096
 SHA512:
-  metadata.gz: df4b845d6a07058d0b4584e713bc6dc5c9e906a4daa3e0d150bb38791da3c677954ceda25aec0e5714b7d2752fa9c2d9291abd64e414adc22b3dbb6ea952c8e9
-  data.tar.gz: f95acb679e150ddaff4035f26fb0f73d1c06db1b6659f0c523d53ed7f35ecaca3c2d6609d0cb9d3683b117f77a13a05030cecb6f51efc4a9de2e51428f272216
+  metadata.gz: '08fe3f4f91088d4fbaa616435c20fb6dce0f4d3dbc0b2912bfd31fd0bf779b0cf80176d743b976e45f461683d46e0018a9c9515f28b9cb5e71b5be59e7fe5770'
+  data.tar.gz: 8fcab338010c3c54af5befa679cfa6c6bfe1c480f6cbd62e1574039e7c0f368d39335078ea0dcb68de5978f6c72984b7f207379b9250e9e6359d659314df756e

data/ext/manasimu/ford_fulkerson.cpp

@@ -0,0 +1,125 @@
+#include "./config.h"
+#include <bits/stdc++.h>
+using namespace std;
+
+#ifndef NO_RUBY_EXT
+#include <rice/rice.hpp>
+#include <rice/stl.hpp>
+using namespace Rice;
+#endif
+
+template <class T>
+struct Edge {
+  int rev, from, to;  // rev:逆向きの辺の番号
+  T cap, original_cap;
+  Edge(int r, int f, int t, T c) : rev(r), from(f), to(t), cap(c), original_cap(c) {}
+};
+
+template <class T>
+struct Graph {
+  vector<vector<Edge<T>>> G;
+  Graph(int n = 0) : G(n) {}
+  vector<Edge<T>>& operator[](int i) { return G[i]; }
+  const size_t size() const { return G.size(); }
+  Edge<T>& redge(Edge<T> e) {  // 逆向きの辺を返す
+    return G[e.to][e.rev];   // 自己ループはないと仮定（あれば G[e.to][e.rev + 1] とする必要がある）
+  }
+  void add_edge(int from, int to, T cap) {  // 有向辺を加える
+    G[from].push_back(Edge<T>((int)G[to].size(), from, to, cap));
+    G[to].push_back(Edge<T>((int)G[from].size() - 1, to, from, 0));
+  }
+};
+
+/* FordFulkerson: Ford-Fulkersonのアルゴリズムで最大流を求める構造体
+   max_flow(G,s,t) ：グラフGの最大流が求まる
+   副作用：G は最大流の残余ネットワークになる
+   計算量: O(|f*||E|) (f*:最大流) 
+ */
+template <class T>
+struct FordFulkerson {
+  const T INF = 1e9;
+  vector<int> used;
+  vector<int> last_used;
+
+  FordFulkerson(){};
+  T dfs(Graph<T>& G, int v, int t, T f) {  // 増加可能経路を見つけて増加分のフローを返す
+    if (v == t) return f;
+    used[v] = 1;
+    for (auto& e : G[v]) {
+      if (used[e.to] == 0 && e.cap > 0) {
+        T d = dfs(G, e.to, t, min(f, e.cap));
+        if (d > 0) {
+          e.cap -= d;
+          G.redge(e).cap += d;
+          return d;
+        }
+      }
+    }
+    return 0;
+  }
+  T max_flow(Graph<T>& G, int s, int t) {
+    T flow = 0;
+    while (true) {
+      last_used.clear();
+      for (int i = 0; i < used.size(); i++) {
+        last_used.push_back(used[i]);
+      }
+      used.assign(G.size(), 0);
+      T f = dfs(G, s, t, INF);  // f が増加分のフロー
+      if (f == 0) {
+        return flow;
+      } else {
+        flow += f;
+      }
+    }
+    return 0;
+  }
+};
+
+#ifdef NO_RUBY_EXT
+int main() {
+  int X, Y, E;
+  cin >> X >> Y >> E;
+  Graph<int> G(X + Y + 2);
+  for (int i = 0; i < X; i++) {  // ソース(0)から X(1~X) への辺
+    G.add_edge(0, i + 1, 1);
+  }
+  for (int i = 0; i < E; i++) {  // X(1~X) から Y(X+1~X+Y) への辺
+    int x, y;
+    cin >> x >> y;
+    G.add_edge(x + 1, y + X + 1, 1);
+  }
+  for (int i = 0; i < Y; i++) {  // Y(X+1~X+Y) からシンク(X+Y+1)への辺
+    G.add_edge(i + X + 1, X + Y + 1, 1);
+  }
+
+  FordFulkerson<int> ff;
+  cout << ff.max_flow(G, 0, X + Y + 1) << endl;
+}
+#else
+
+extern "C" {
+  
+  void Init_ford_fulkerson() {
+
+    Data_Type<Edge<int>> rb_cEdge =
+      define_class<Edge<int>>("Edge")
+      .define_attr("rev", &Edge<int>::rev, Rice::AttrAccess::Read)
+      .define_attr("from", &Edge<int>::from, Rice::AttrAccess::Read)
+      .define_attr("to", &Edge<int>::to, Rice::AttrAccess::Read)
+      .define_attr("cap", &Edge<int>::cap, Rice::AttrAccess::Read);
+
+    Data_Type<Graph<int>> rb_cGraph =
+      define_class<Graph<int>>("Graph")
+      .define_constructor(Constructor<Graph<int>, int>(), (Arg("n")))
+      .define_attr("G", &Graph<int>::G, Rice::AttrAccess::Read)
+      .define_method("add_edge", &Graph<int>::add_edge);
+
+    Data_Type<FordFulkerson<int>> rb_cFordFulkerson =
+      define_class<FordFulkerson<int>>("FordFulkerson")
+      .define_constructor(Constructor<FordFulkerson<int>>())
+      .define_method("max_flow", &FordFulkerson<int>::max_flow)
+      .define_attr("used", &FordFulkerson<int>::last_used, Rice::AttrAccess::Read);
+  };
+}
+#endif

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: manasimu
 version: !ruby/object:Gem::Version
-  version: 0.0.36
+  version: 0.0.38
 platform: ruby
 authors:
 - so1itaryrove
@@ -18,7 +18,9 @@ extensions:
 extra_rdoc_files: []
 files:
 - db/card_type_aggregate
+- ext/manasimu/config.h
 - ext/manasimu/extconf.rb
+- ext/manasimu/ford_fulkerson.cpp
 - lib/manasimu.rb
 - lib/manasimu/api_util.rb
 - lib/manasimu/card.rb