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

…nemartin#167)
aymns · Sep 13, 2020 · a910b84 · a910b84
1 parent 0cc9362
commit a910b84
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):