#!/usr/bin/env python3 from functools import reduce from re import findall from copy import deepcopy import sys # import numpy as np filename = sys.argv[1] if len(sys.argv) == 2 \ else "in/" + sys.argv[0].split('/')[-1].rstrip(".py") + ".pzl" data = open(filename).read() lines = data.rstrip('\n').split('\n') res1 = 0 res2 = 0 pos1 = 50 pos2 = 50 for line in lines: lr = line[0] lr = 1 if lr == 'R' else -1 n = int(line[1:]) mov = lr * n pos1 += mov pos1 %= 100 if pos1 == 0: res1 += 1 for i in range(0, n): pos2 += lr pos2 %= 100 if pos2 == 0: res2 += 1 print('res1:', res1) print('res2:', res2)