Fix donnemartin#148: Add State(Enum) to social_graph_snippets.py (don…

…nemartin#167)
Godfrey-Ndungu · Jul 15, 2018 · 7b20f5c · 7b20f5c
1 parent ded61ab
commit 7b20f5c
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):