From e8063b640dd5b1a926b215f816d1c64a95925654 Mon Sep 17 00:00:00 2001 From: EtherealEntity Date: Wed, 23 Mar 2022 11:03:54 +1100 Subject: [PATCH] filter up down movement when on a seat --- main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index a859fea..96e04e6 100644 --- a/main.py +++ b/main.py @@ -43,7 +43,11 @@ def nextSquare(passenger, grid): candidati = [] for i in mooreNeighbourhood(passenger.curr): if manhattanDistance(passenger.dest, i) < manhattanDistance(passenger.dest, passenger.curr): - candidati.append(i) + if grid[i[0]][i[1]].typ == SquareType.SEAT: + candidati.append(passenger.curr[0], passenger.curr[1] + 1) + candidati.append(passenger.curr[0], passenger.curr[1] - 1) + else: + candidati.append(i) for i in candidati: if grid[i[0]][i[1]].typ == SquareType.AISLE and grid[i[0]][i[1]].occupant != None: #this should work well enough for one aisle plane return i