View source for 2845 - Wildfire 1

== Rezolvare == 
<syntaxhighlight lang="python" line="1">
# 2845 - Wildfire 1
def validate_date_intrare(x: int, y: int, n: int, m: int) -> bool:
    """
    Verifică dacă datele de intrare sunt valide.
    :param x: coordonata x a copacului de la care a izbucnit focul
    :param y: coordonata y a copacului de la care a izbucnit focul
    :param n: numărul de linii ale pădurii
    :param m: numărul de coloane ale pădurii
    :return: True dacă datele sunt valide, False altfel
    """
    if not (1 <= x <= n <= 1000 and 1 <= y <= m <= 1000):
        return False
    return True


def rezolva_problema(x: int, y: int, n: int, m: int) -> list:
    """
    Calculează numărul de zile după care va fi afectat fiecare copac al pădurii.
    :param x: coordonata x a copacului de la care a izbucnit focul
    :param y: coordonata y a copacului de la care a izbucnit focul
    :param n: numărul de linii ale pădurii
    :param m: numărul de coloane ale pădurii
    :return: lista cu numărul de zile pentru fiecare copac al pădurii
    """
    matrice = [[0] * m for _ in range(n)]
    matrice[x][y] = 1
    for val in range(2, n * m + 1):
        for i, linie in enumerate(matrice):
            for j, elem in enumerate(linie):
                if elem == val - 1:
                    for ii, jj in ((i + 1, j), (i - 1, j), (i, j + 1), (i, j - 1)):
                        if 0 <= ii < n and 0 <= jj < m and matrice[ii][jj] == 0:
                            matrice[ii][jj] = val
    return [elem - 1 for linie in matrice for elem in linie]


if __name__ == '__main__':
    try:
        x, y, n, m = map(int, input().strip().split())
        if not validate_date_intrare(x, y, n, m):
            print("Datele nu corespund restricțiilor impuse.")
        else:
            rezultat = rezolva_problema(x, y, n, m)
            print("Datele sunt introduse corect.")
            for i in range(n):
                print(*rezultat[i * m: (i + 1) * m])
    except ValueError:
        print("Datele introduse sunt invalide.")







</syntaxhighlight>