Browse Source

Fix #148: Add State(Enum) to social_graph_snippets.py (#167)

cclauss 6 years ago
parent
commit
e116bd75fc
1 changed files with 6 additions and 0 deletions
  1. 6 0
      solutions/system_design/social_graph/social_graph_snippets.py

+ 6 - 0
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):