Editing 3327 - La Scoala

== Cerinţa ==
Directorul unei şcoli doreşte să premieze la sfârşitul anului şcolar pe cei mai buni elevi la învăţătură. Pentru acest lucru el are de rezolvat două probleme:

1. Să determine câţi elevi vor fi premiaţi dintre cei '''n''' ( '''2≤n≤700''' ) elevi ai şcolii. După discuţii aprinse cu ceilalţi profesori se hotărăşte în Consiliul Profesoral ca numărul premianţilor să fie '''n-k''' , unde '''k''' este cel mai mare număr pătrat perfect mai mic strict decât '''n''' . De exemplu, pentru '''n=150''' , '''k''' este '''144''' (pentru că '''144=12*12''' ), deci vor fi premiaţi '''150'''- '''144''' = '''6''' elevi.
2. Pentru a fi cât mai multă linişte la premiere, în Consiliul Profesoral se ia decizia ca elevii care nu vor fi premiaţi să fie aşezaţi pe terenul de sport pe rânduri de câte '''p''' elevi (unde '''p*p=k''' ). În acest scop, directorul a numerotat elevii nepremiaţi de la 1 la k şi a hotărât ca elevii să fie aşezaţi în ordinea descrescătoare a numerelor asociate.

Scrieți un program care citește numărul de elevi din școală și afișează numărul de elevi premiați și apoi aranjarea elevilor nepremiați.
== Date de intrare ==
Programul citește de la tastatură numărul n reprezentând numărul de elevi din școală.
== Date de ieşire ==
Programul va afișa pe ecran numărul de elevi premiaţi și pe urmatoarele k linii o matrice kxk reprezentând modul în care vor fi așezaţi elevii nepremiaţi.
== Restricții și precizări ==
* 1 &les n &les 700
== Exemplu ==
; Intrare
: 35
; Iesire
: 10
: 25 24 23 22 21 
: 20 19 18 17 16 
: 15 14 13 12 11 
: 10 9 8 7 6 
: 5 4 3 2 1
<syntaxhighlight lang="python" line>
import math

# Functia premiati calculeaza numarul de elevi premiati si k
def premiati(n):
    # k este cel mai mare numar patrat perfect mai mic strict decat n
    k = int(math.sqrt(n))**2
    # Numarul de premianti este n - k
    premianti = n - k
    return premianti, k

# Functia aranjare creeaza o matrice k x k cu elevii nepremiati
def aranjare(k):
    # Numarul de elevi pe rand este radacina patrata a lui k
    p = int(math.sqrt(k))
    # Initializam matricea cu 0
    elevi = [[0]*p for _ in range(p)]
    # Numarul asociat fiecarui elev incepe de la k si scade cu 1 la fiecare pas
    numar = k
    for i in range(p):
        for j in range(p):
            elevi[i][j] = numar
            numar -= 1
    return elevi

# Functia afisare printeaza matricea pe ecran
def afisare(elevi):
    for rand in elevi:
        print(' '.join(map(str, rand)))

# In main citim numarul de elevi, calculam numarul de premianti si aranjam elevii nepremiati
if __name__ == "__main__":
    n = int(input())
    premianti, k = premiati(n)
    print(premianti)
    elevi = aranjare(k)
    afisare(elevi)

</syntaxhighlight>