Задание 3. Дан двумерный массив А размерности n x m, заполненный случайными целыми числами. Напишите программу на языке Python для нахождения суммы элементов, расположенных на главной диагонали (элементы с одинаковыми индексами)
Ответы
Ответ:
Код на Python для нахождения суммы элементов главной диагонали двумерного массива:
import random
# определяем размерность массива
n = 4
m = 4
# создаем массив и заполняем его случайными целыми числами
A = [[random.randint(1, 9) for j in range(m)] for i in range(n)]
# выводим массив на экран
print("Массив A:")
for row in A:
print(row)
# находим сумму элементов на главной диагонали
sum_diag = 0
for i in range(n):
sum_diag += A[i][i]
# выводим сумму на экран
print("Сумма элементов на главной диагонали равна", sum_diag)
Пример вывода:
Массив A:
[6, 9, 7, 8]
[6, 1, 1, 7]
[1, 6, 8, 2]
[9, 9, 3, 8]
Сумма элементов на главной диагонали равна 23
В этом примере мы используем вложенные циклы для перебора всех элементов массива и нахождения суммы элементов на главной диагонали. Внешний цикл перебирает строки массива, а внутренний цикл перебирает столбцы. Если индексы строки и столбца совпадают, то мы складываем элемент с суммой элементов на главной диагонали.