C_get_neighbours.py

"""
C. Соседи
Дана матрица. Нужно написать функцию, которая для элемента возвращает всех его
соседей. Соседним считается элемент, находящийся от текущего на одну ячейку
влево, вправо, вверх или вниз. Диагональные элементы соседними не считаются.

Например, в матрице A соседними элементами для (0, 0) будут 2 и 0.
А для (2, 1) –— 1, 2, 7, 7.

Формат ввода
В первой строке задано n — количество строк матрицы. Во второй — количество
столбцов m. Числа m и n не превосходят 1000. В следующих n строках задана
матрица. Элементы матрицы — целые числа, по модулю не превосходящие 1000.
В последних двух строках записаны координаты элемента, соседей которого
нужно найти. Индексация начинается с нуля.

Формат вывода
Напечатайте нужные числа в возрастающем порядке через пробел.
Пример 1
Ввод
4
3
1 2 3
0 2 6
7 4 1
2 7 0
3
0
Вывод
7 7

"""

from typing import List, Tuple

from ..timing import timing


@timing
def get_neighbours(matrix: List[List[int]], row: int, col: int) -> List[int]:
    h = []
    if col + 1 < len(matrix[0]):
        h.append(matrix[row][col + 1])
    if col - 1 >= 0:
        h.append(matrix[row][col - 1])
    if row + 1 < len(matrix):
        h.append(matrix[row + 1][col])
    if row - 1 >= 0:
        h.append(matrix[row - 1][col])
    return h


@timing
def read_input() -> Tuple[List[List[int]], int, int]:
    n = int(input())
    _ = int(input())
    matrix = []
    for _ in range(n):
        matrix.append(list(map(int, input().strip().split())))
    row = int(input())
    col = int(input())
    return matrix, row, col


if __name__ == "__main__":
    matrix, row, col = read_input()
    print(*sorted(get_neighbours(matrix, row, col)))