GitHub/interactive-coding-challenges
1
0
Fork 0
mirror of https://github.com/donnemartin/interactive-coding-challenges synced 2026-01-06 17:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added three state variable visit_state which will be useful for more advanced challenges such as topological sort.

Browse Source
This commit is contained in:
Donne Martin
2015-08-23 08:08:46 -04:00
parent 82b74ca7b4
commit 1f45095f60
2 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
graphs_trees/graph/graph_solution.ipynb
View File

@@ -135,13 +135,20 @@
"source": [
"%%writefile graph.py\n",
"from collections import OrderedDict\n",
"from enum import Enum # Python 2 users: Run pip install enum34\n",
"\n",
"\n",
"class State(Enum):\n",
" unvisited = 1\n",
" visited = 2\n",
" visiting = 3\n",
"\n",
"\n",
"class Node:\n",
"\n",
" def __init__(self, id):\n",
" self.id = id\n",
" self.visited = False\n",
" self.visit_state = State.unvisited\n",
" self.adjacent = OrderedDict() # key = node, val = weight\n",
"\n",
" def __str__(self):\n",