From 72f5124cf4a3da281dedf332d6bca5116f8e6b49 Mon Sep 17 00:00:00 2001
From: Abdurrezzak Efe <abdurrezak.efe@ug.bilkent.edu.tr>
Date: Tue, 28 Nov 2017 21:33:56 +0300
Subject: [PATCH] Update 13.DFS.cpp

---
 13.DFS.cpp | 32 +++++++++++++++++++-------------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/13.DFS.cpp b/13.DFS.cpp
index 4288a70..25bcd82 100644
--- a/13.DFS.cpp
+++ b/13.DFS.cpp
@@ -16,7 +16,7 @@ using namespace std;
 class Graph{
 
     vector<vector<int>> edge;
-    vector<int> color; // 0 black(finished node), 1 white(neither visited nor finished node)
+    vector<int> color; // 1 black(finished node), 0 white(neither visited nor finished node)
 
 public:
 
@@ -24,20 +24,21 @@ class Graph{
     {
         edge.resize(v);
         color.resize(v);
-        for(int i=0;i<v;i++) //initializing nodes' colors
-            color[i] = 1;
+        for(int i=0;i<edge.size();i++) //initializing nodes' colors
+            color[i] = 0;
     }
 
-    void add_edge(int v, int u)
+    void add_edge(int u, int v)
     {
         //adding from both directions makes it undirected
-        edge[v].push_back(u);
+        //edge[v].push_back(u);
         edge[u].push_back(v);
     }
 
     void DFS(int s) //dfs operation
     {
-        if(!color[s])
+
+        if(color[s])
             return;
 
         stack<int> st;
@@ -46,12 +47,16 @@ class Graph{
         while(!st.empty())
         {
             s = st.top();
-            if(color[s] == 0)
-                continue;
-            color[s] = 0, cout << s+1 <<  " ";
             st.pop();
+
+            if(!color[s])
+            {
+                cout << s << " ";
+                color[s] = 1;
+            }
+
             for(int i=0;i<edge[s].size();i++)
-                if(color[edge[s][i]] == 1)
+                if(color[edge[s][i]] == 0)
                     st.push(edge[s][i]);
         }
     }
@@ -74,7 +79,8 @@ int main()
     for(int i=0;i<e;i++)
         cin >> u >> w, g.add_edge(u,w);
 
+    cout << endl;
 
-    for(int i=0;i<v;i++) //this for prevents our dfs from not traversing on non-connected parts
-        g.DFS(i); //starting DFS from vertex i
-}
\ No newline at end of file
+    for(int i=0;i<v;i++)
+        g.DFS(i);
+}