xiretza/advent-of-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

day7/python: rename functions

Browse source
This commit is contained in:
Xiretza 2020-12-07 07:20:34 +01:00
parent 3566e499b6
commit 9b12fb9d02
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
2020/day7/day7.py
View file

@ -27,24 +27,24 @@ def process_rules(rules):
return rules_dict return rules_dict
def can_contain(rules, colour): def parents_of(rules, colour):
parents = set() parents = set()
for parent, children in rules.items(): for parent, children in rules.items():
if colour in children: if colour in children:
parents.add(parent) parents.add(parent)
parents |= can_contain(rules, parent) parents |= parents_of(rules, parent)
return parents return parents
def colour_contains(rules, colour): def children_of(rules, colour):
children = defaultdict(lambda: 0) children = defaultdict(lambda: 0)
for child_colour, child_num in rules[colour].items(): for child_colour, child_num in rules[colour].items():
children[child_colour] += child_num children[child_colour] += child_num
for col, num in colour_contains(rules, child_colour).items(): for col, num in children_of(rules, child_colour).items():
children[col] += child_num * num children[col] += child_num * num
return children return children
if __name__ == '__main__': if __name__ == '__main__':
bag_rules = [rule.strip() for rule in sys.stdin.readlines()] bag_rules = [rule.strip() for rule in sys.stdin.readlines()]
rules = process_rules(bag_rules) rules = process_rules(bag_rules)
print(len(can_contain(rules, 'shiny gold'))) print(len(parents_of(rules, 'shiny gold')))
print(sum(colour_contains(rules, 'shiny gold').values())) print(sum(children_of(rules, 'shiny gold').values()))