a.py (297B)
1 import fileinput 2 3 with fileinput.input() as lines: 4 rows = [line[:-1] for line in lines] 5 6 s = 0 7 pos = [rows[0].find('S')] 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)