View source for 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 &les; n &les; 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 ==
<syntaxhighlight lang="python" line>

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.")


</syntaxhighlight>