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

<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):
       row = input().split()
       for j in range(1, n + 1):
           a[i][j] = int(row[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>