From e116bd75fca589f0201fae062f94bf991298cd7b Mon Sep 17 00:00:00 2001 From: cclauss Date: Sun, 15 Jul 2018 02:01:57 +0200 Subject: [PATCH] Fix #148: Add State(Enum) to social_graph_snippets.py (#167) --- .../system_design/social_graph/social_graph_snippets.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/solutions/system_design/social_graph/social_graph_snippets.py b/solutions/system_design/social_graph/social_graph_snippets.py index 8a83f74..f0ea4c4 100644 --- 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):