62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
|
import sys, os
|
||
|
|
import numpy
|
||
|
|
import time
|
||
|
|
|
||
|
|
class display:
|
||
|
|
def __init__(self):
|
||
|
|
self.w, self.h = tuple(os.get_terminal_size())
|
||
|
|
#self.h -= 1
|
||
|
|
border = "o"
|
||
|
|
self.delta = numpy.array((self.w//2, self.h//2))
|
||
|
|
top = border * (self.w)
|
||
|
|
line = border + " " * (self.w - 2) + border
|
||
|
|
self.background = top + "".join(
|
||
|
|
line for _ in range(self.h - 2)) + top
|
||
|
|
|
||
|
|
def __call__(self, knots):
|
||
|
|
print("\033[0;0H", end='')
|
||
|
|
x, y = self.delta + knots[0]
|
||
|
|
J = 20
|
||
|
|
if x <= 0:
|
||
|
|
self.delta[0] += J
|
||
|
|
if x >= self.w-1:
|
||
|
|
self.delta[0] -= J
|
||
|
|
if y <= 0:
|
||
|
|
self.delta[1] += J
|
||
|
|
if y >= self.h-1:
|
||
|
|
self.delta[1] -= J
|
||
|
|
txt = [c for c in self.background]
|
||
|
|
for i,knot in enumerate(reversed(knots)):
|
||
|
|
x, y = knot + self.delta
|
||
|
|
txt[int(x + y * self.w)] = str(9-i)
|
||
|
|
print("".join(txt),end='')
|
||
|
|
time.sleep(.008)
|
||
|
|
|
||
|
|
d = display()
|
||
|
|
|
||
|
|
def input():
|
||
|
|
for l in sys.stdin:
|
||
|
|
m, q = l.split()
|
||
|
|
m = dict(U=(0,1), D=(0,-1), L=(-1,0), R=(1, 0))[m]
|
||
|
|
for _ in range(int(q)):
|
||
|
|
yield m
|
||
|
|
|
||
|
|
def follow(head, tail):
|
||
|
|
delta = head - tail
|
||
|
|
if max(abs(delta)) <= 1: # no move
|
||
|
|
return
|
||
|
|
if delta[0]:
|
||
|
|
tail[0] += 1 if delta[0] > 0 else -1
|
||
|
|
if delta[1]:
|
||
|
|
tail[1] += 1 if delta[1] > 0 else -1
|
||
|
|
|
||
|
|
knots = [numpy.array((0, 0)) for _ in range(10)]
|
||
|
|
visits = [ {(0,0)} for _ in range(10)]
|
||
|
|
for direction in input():
|
||
|
|
knots[0] += direction
|
||
|
|
for n in range(1, 10):
|
||
|
|
follow(knots[n-1], knots[n])
|
||
|
|
d(knots)
|
||
|
|
visits[n].add(tuple(knots[n]))
|
||
|
|
print(" ".join(str(len(v)) for v in visits))
|