Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- DIRS = [(0, -1), (1, 0), (0, 1), (-1, 0)]
- class AntSim(object):
- def __init__(self):
- self.grid = [[0 for _ in xrange(20)] for _ in xrange(20)]
- self.antx = 10
- self.anty = 10
- self.antd = 0
- def advancePrint(self):
- vals = ["_", "#", "A"]
- current = self.grid[self.anty][self.antx]
- self.grid[self.anty][self.antx] = 2
- for row in self.grid:
- print ''.join(vals[i] for i in row)
- if current == 0:
- self.grid[self.anty][self.antx] = 1
- self.antd += 1
- if self.antd > 3:
- self.antd = 0
- elif current == 1:
- self.grid[self.anty][self.antx] = 0
- self.antd -= 1
- if self.antd < 0:
- self.antd = 3
- self.antx += DIRS[self.antd][0]
- self.anty += DIRS[self.antd][1]
- a = AntSim()
- while True:
- raw_input()
- print
- a.advancePrint()
Add Comment
Please, Sign In to add comment