Add State(Enum) to social_graph_snippets.py

* Fixes donnemartin#148 * Enables donnemartin#93 * Replaces donnemartin#162
cclauss · Jul 13, 2018 · b646b87 · b646b87
1 parent 9069993
commit b646b87
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/solutions/system_design/social_graph/social_graph_snippets.py b/solutions/system_design/social_graph/social_graph_snippets.py
@@ -1,5 +1,11 @@
 # -*- coding: utf-8 -*-
 from collections import deque
+from enum import Enum
+
+
+class State(Enum):
+    unvisited = 0
+    visited = 1
 
 
 class Graph(object):