• + 0 comments

    I employed a very similar method:

    def cavityMap(grid):
    if n < 3:
        return grid
    else:
        for idx, row in enumerate(grid[1:-1]):
            for jdx, i in enumerate(row[1:-1]):
                if all([grid[idx][jdx+1]<i, grid[idx+2][jdx+1]<i, 
                row[jdx] < i , row[jdx+2] < i]):
                    grid[idx+1] = grid[idx+1][:jdx+1] + 'X' + grid[idx+1][jdx+2:]
        return grid