Algorithms : complete 03-algorithms_on_graphs 03-paths_in_graphs

1 files changed, 26 insertions, 3 deletions

diff --git a/03-algorithms_on_graphs/03-paths_in_graphs/02-bipartite/bipartite.cpp b/03-algorithms_on_graphs/03-paths_in_graphs/02-bipartite/bipartite.cpp
index f97eca5..dfd74e1 100644
--- a/03-algorithms_on_graphs/03-paths_in_graphs/02-bipartite/bipartite.cpp
+++ b/03-algorithms_on_graphs/03-paths_in_graphs/02-bipartite/bipartite.cpp
@@ -6,8 +6,31 @@ using std::vector;
 using std::queue;
 
 int bipartite(vector<vector<int> > &adj) {
-  //write your code here
-  return -1;
+  vector<int> t(adj.size(), -1);
+
+  queue<int> q;
+  q.push(0);
+  t[0] = 0;
+
+  int tag = 1;
+  while (!q.empty()) {
+      int v = q.front();
+      /* std::cout << " check : " << v << " tag " << tag << std::endl; */
+      q.pop();
+      tag = !t[v];
+      for (int i = 0; i < adj[v].size(); i++) {
+          int w = adj[v][i];
+          /* std::cout << "   " << w  << " " << t[w] << std::endl; */
+          if (t[w] == -1) {
+              q.push(w);
+              t[w] = tag;
+          } else if (t[w] != tag) {
+              /* std::cout << "   wrong " << w  << " " << t[w] << std::endl; */
+              return 0;
+          }
+      }
+  }
+  return 1;
 }
 
 int main() {
@@ -20,5 +43,5 @@ int main() {
     adj[x - 1].push_back(y - 1);
     adj[y - 1].push_back(x - 1);
   }
-  std::cout << bipartite(adj);
+  std::cout << bipartite(adj) << '\n';
 }