make Dict default vertexcolormap

sbromberger · Apr 12, 2016 · 7d5e2d9 · 7d5e2d9
1 parent d55b136
commit 7d5e2d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/traversals/bfs.jl b/src/traversals/bfs.jl
@@ -48,7 +48,7 @@ function traverse_graph!(
     alg::BreadthFirst,
     source,
     visitor::SimpleGraphVisitor;
-    vertexcolormap = zeros(Int, nv(graph)),
+    vertexcolormap = Dict{Int, Int}(),
     edgecolormap = DummyEdgeMap(),
     queue = Vector{Int}())
 
@@ -224,7 +224,7 @@ BipartiteVisitor(n::Int) = BipartiteVisitor(zeros(UInt8,n), true)
 
 function examine_neighbor!(visitor::BipartiteVisitor, u::Int, v::Int, vcolor::Int, ecolor::Int)
     if vcolor == 0
-        visitor.bipartitemap[v] = (visitor.bipartitemap[u] == 1)? 2:1
+        visitor.bipartitemap[v] = (visitor.bipartitemap[u] == 1) ? 2 : 1
     else
         if visitor.bipartitemap[v] == visitor.bipartitemap[u]
             visitor.is_bipartite = false

diff --git a/src/traversals/dfs.jl b/src/traversals/dfs.jl
@@ -58,7 +58,7 @@ function traverse_graph!(
     alg::DepthFirst,
     s::Int,
     visitor::SimpleGraphVisitor;
-    vertexcolormap = zeros(Int, nv(graph)),
+    vertexcolormap = Dict{Int, Int}(),
     edgecolormap = DummyEdgeMap())
 
     vertexcolormap[s] = 1