aoc

My solutions for the Advent of Code
git clone https://git.tronto.net/aoc
Download | Log | Files | Refs | README

a.py (298B)

      1 import fileinput
      2 
      3 with fileinput.input() as lines:
      4 	rows = [line[:-1] for line in lines]
      5 	pos = [rows[0].find('S')]
      6 
      7 s = 0
      8 for row in rows[1:]:
      9 	newpos = set()
     10 	for p in pos:
     11 		if row[p] == '^':
     12 			s += 1
     13 			newpos.add(p-1)
     14 			newpos.add(p+1)
     15 		else:
     16 			newpos.add(p)
     17 	pos = list(newpos)
     18 
     19 print(s)