From b646b87d6c1c15e0372d6af0ae5c317cef7ed653 Mon Sep 17 00:00:00 2001 From: cclauss Date: Fri, 13 Jul 2018 13:58:40 +0200 Subject: [PATCH] Add State(Enum) to social_graph_snippets.py * Fixes #148 * Enables #93 * Replaces #162 --- .../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 8a83f748..f0ea4c4b 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):