0794 - Serpuire

Cerinţa

Se dă o matrice cu n linii şi n coloane şi elemente numere naturale. Să se afișeze elementele prin parcurgerea șerpuită a matricei, începând din elementul de pe prima linie și prima coloană, ca în exemplu.

Date de intrare

Programul citește de la tastatură numărul n, iar apoi n*n numere naturale, separate prin spaţii, reprezentând elementele matricei, linie cu linie.

Date de ieşire

Dacă datele sunt introduse corect,pe ecran se va afișa :"Datele sunt introduse corect.",apoi pe un rând nou elementele din matrice în ordinea cerută, separate prin câte un spațiu.În cazul contrar,se va afișa pe ecran "Datele nu corespund restricțiilor impuse.".

Restricții și precizări

• 1 ⩽ n ⩽ 25
• elementele matricei vor fi mai mici decât 1.000.000

Exemplu

Date de intrare

4

1 3 4 10

2 5 9 11

6 8 12 15

7 13 14 16

Date de ieșire

Datele sunt introduse corect.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

Rezolvare

def validare_date(n, a):
    # Verificăm dacă n se află în intervalul dat
    if n < 1 or n > 25:
        return False

    # Verificăm dacă toate elementele matricei sunt mai mici decât 1.000.000
    for i in range(1, n + 1):
        for j in range(1, n + 1):
            if a[i][j] >= 1000000:
                return False

    # Dacă n și elementele matricei sunt valide, returnăm True
    return True


def șerpuire(n, a):
    rezultat = []

    # Parcurgem matricea în modul cerut
    for i in range(1, n + 1):
        if i % 2 == 0:  # in jos
            for j in range(i, 0, -1):
                rezultat.append(str(a[j][i - j + 1]))
        else:  # în sus
            for j in range(i, 0, -1):
                rezultat.append(str(a[i - j + 1][j]))

    for j in range(2, n + 1):
        if j % 2 != n % 2:  # in jos
            for i in range(n, j - 1, -1):
                rezultat.append(str(a[n - i + j - 1][i]))
        else:  # în sus
            for i in range(n, j - 1, -1):
                rezultat.append(str(a[i][n - i + j - 1]))

    # Returnăm rezultatul sub forma de șir de caractere cu elementele separate prin spații
    return " ".join(rezultat)


if __name__ == '__main__':
    # Citim datele de intrare
    n = int(input())
    a = [[0 for j in range(n + 1)] for i in range(n + 1)]
    for i in range(1, n + 1):
        rând = input().split()
        for j in range(1, n + 1):
            a[i][j] = int(rând[j - 1])

    # Verificăm dacă datele sunt valide și afișăm un mesaj corespunzător
    if validare_date(n, a):
        print("\nDatele sunt introduse corect.\n")
        # Rezolvăm problema și afișăm rezultatul
        print(șerpuire(n, a))
    else:
        print("Datele nu corespund restrictiilor impuse.")