1772 - Progresie 2

Cerința

Se consideră o progresie aritmetică cu primul termen a1 și rația r, precum și un număr n. Să se afișeze o matrice pătratică cu n linii și n coloane care să conțină termenii acestei progresii astfel:

• prima coloană va conține primii n termeni, în ordine, de sus în jos
• a doua coloană va conține următorii n termeni, în ordine, de sus în jos etc.

Date de intrare

Programul citește de la tastatură numerele a1, r, n.

Date de ieșire

Programul va afișa pe ecran elementele matricei, câte o linie a matricei pe o linei a ecranului, elementele de pe fiecare linie fiind separate prin câte un spațiu.

Restricții și precizări

• 1 ⩽ n ⩽ 200
• elementele matricei vor fi numere întregi ce pot fi reprezentate pe 4 octeți cu semn.

Exemplu 1

Intrare

-7 3 4

Ieșire

-7 5 17 29

-4 8 20 32

-1 11 23 35

2 14 26 38

Exemplu 2

Intrare

0

0

0

Ieșire

Nu au fost respectate cerintele impuse

Rezolvare

#1772 - Progresie2
def is_valid_input(a1, r, n):
    return 1 <= n <= 200 and isinstance(a1, int) and isinstance(r, int)

def build_matrix(a1, r, n):
    matrix = [[0] * n for _ in range(n)]

    for j in range(n):
        for i in range(n):
            matrix[i][j] = a1 + i * r + j * r * n

    return matrix

def print_matrix(matrix):
    for row in matrix:
        print(" ".join(map(str, row)))

if __name__ == "__main__":
    a1 = int(input("Introduceți primul termen a1: "))
    r = int(input("Introduceți rația r: "))
    n = int(input("Introduceți numărul n: "))

    if is_valid_input(a1, r, n):
        result_matrix = build_matrix(a1, r, n)
        print_matrix(result_matrix)
    else:
        print("Nu au fost respectate cerintele impuse.")