Guest User

Untitled

a guest
Feb 21st, 2018
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.35 KB | None | 0 0
  1. def possiblemoves(grid):
  2. h = len(grid)
  3. w = len(grid[0])
  4. m = [] # moves
  5.  
  6. for i, idata in enumerate(grid):
  7. for j, jdata in enumerate(idata):
  8. # Northwest
  9. if (i-2) > 0 and (j-2) > 0 and grid[i][j] == "P" and grid[i-1][j-1] == "P" and grid[i-2][j-2] == "H": m.append(((i,j), (i-1,j-1), (i-2, j-2)))
  10. # Southwest
  11. if (i+2) < h and (j-2) > 0 and grid[i][j] == "P" and grid[i+1][j-1] == "P" and grid[i+2][j-2] == "H": m.append(((i,j), (i+1,j-1), (i+2, j-2)))
  12. # Northeast
  13. if (i-2) > 0 and (j+2) < w and grid[i][j] == "P" and grid[i-1][j+1] == "P" and grid[i-2][j+2] == "H": m.append(((i,j), (i-1,j+1), (i-2, j+2)))
  14. # Southeast
  15. if (i+2) < h and (j+2) < w and grid[i][j] == "P" and grid[i+1][j+1] == "P" and grid[i+2][j+2] == "H": m.append(((i,j), (i+1,j+1), (i+2, j+2)))
  16. # East
  17. if (j+4) < h and grid[i][j] == "P" and grid[i][j+2] == "P" and grid[i][j+4] == "H": m.append(((i,j), (i,j+2), (i,j+4)))
  18. # West
  19. if (j-4) > 0 and grid[i][j] == "P" and grid[i][j-2] == "P" and grid[i][j-4] == "H": m.append(((i,j), (i,j-2), (i,j-4)))
  20.  
  21. return m
Add Comment
Please, Sign In to add comment