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