sebse/python-aoc-2020
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Day 7 (possibility to disable dynamic programming)

Browse Source
This commit is contained in:
Sebastian Seedorf
2020-12-07 13:56:08 +01:00
parent e242fec2d9
commit dafba285ad
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
day07/part2.py
View File

@@ -1,15 +1,18 @@
from day07.common import BaseGraph from day07.common import BaseGraph
import datetime
class Graph(BaseGraph): class Graph(BaseGraph):
def depth_first_search(self, start: str, search_contained_by: bool = False): def depth_first_search(self, start: str, search_contained_by: bool = False, use_cache=True):
visited = [] visited = []
def count_bags(nxt): def count_bags(nxt):
if nxt not in self._nodes: if nxt not in self._nodes:
return 0 return 0
node = self._nodes[nxt] node = self._nodes[nxt]
if not use_cache:
node.visited = False
node.count = 0
if node.visited: if node.visited:
return node.count return node.count
node.visited = True node.visited = True
@@ -33,6 +36,13 @@ count = -1
for line in lines: for line in lines:
graph.add_line(line) graph.add_line(line)
a = datetime.datetime.now()
count = graph.depth_first_search("shiny gold", search_contained_by=False) - 1 count = graph.depth_first_search("shiny gold", search_contained_by=False) - 1
print("With dyn programming", datetime.datetime.now() - a)
print(count) print(count)
a = datetime.datetime.now()
count = graph.depth_first_search("shiny gold", search_contained_by=False, use_cache=False) - 1
print("Without dyn programming", datetime.datetime.now() - a)
print(count)