public/aoc2021
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

day12: defaultdicts ... TIL

Browse Source
This commit is contained in:
Stefan Harmuth 2021-12-12 07:17:21 +01:00
parent 457be373ff
commit 86745e1d2c
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
day12.py
View File

@ -1,19 +1,13 @@
from collections import defaultdict
from tools.aoc import AOCDay from tools.aoc import AOCDay
from typing import Any, Dict, List from typing import Any, Dict, List
def getCaveMap(lines: List[List[str]]) -> Dict[str, List[str]]: def getCaveMap(lines: List[List[str]]) -> Dict[str, List[str]]:
m = {} m = defaultdict(list)
for l in lines: for l, r in lines:
if l[0] in m: m[l].append(r)
m[l[0]].append(l[1]) m[r].append(l)
else:
m[l[0]] = [l[1]]
if l[1] in m:
m[l[1]].append(l[0])
else:
m[l[1]] = [l[0]]
return m return m