summaryrefslogtreecommitdiff
path: root/2016/1b.py
diff options
context:
space:
mode:
Diffstat (limited to '2016/1b.py')
-rwxr-xr-x2016/1b.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/2016/1b.py b/2016/1b.py
new file mode 100755
index 0000000..cc24f76
--- /dev/null
+++ b/2016/1b.py
@@ -0,0 +1,15 @@
+#!/usr/bin/env python3
+import sys, re
+
+ins=[({"R":-1j,"L":1j}[s[0]],int(s[1:])) for s in re.findall(r"[RL]\d+",sys.stdin.readline())]
+hits={}
+p=0
+d=1
+for i in ins:
+ d*=i[0]
+ for _ in range(i[1]):
+ p+=d
+ hits[p]=hits.get(p,0)+1
+ if hits[p]>=2:
+ print(int(abs(p.real)+abs(p.imag)))
+ sys.exit(0)