From 01c2cd358b3f693181fa86e82414b3e32f9a890e Mon Sep 17 00:00:00 2001 From: Sebastiano Tronto Date: Sun, 7 Dec 2025 14:28:02 +0100 Subject: Use defaultdict --- 2025/07/b.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to '2025/07/b.py') diff --git a/2025/07/b.py b/2025/07/b.py index 4457654..0a8672e 100644 --- a/2025/07/b.py +++ b/2025/07/b.py @@ -1,15 +1,15 @@ import fileinput +from collections import defaultdict with fileinput.input() as lines: rows = [line[:-1] for line in lines] pos = {rows[0].find('S'): 1} for row in rows[1:]: - newpos = {} + newpos = defaultdict(int) for p, w in pos.items(): for pp in [p-1,p+1] if row[p] == '^' else [p]: - t = newpos.get(pp, 0) - newpos[pp] = t + w + newpos[pp] += w pos = dict(newpos) print(sum(pos.values())) -- cgit v1.3