mirror of
https://github.com/autistic-symposium/master-algorithms-py.git
synced 2025-04-29 20:26:07 -04:00
Add files via upload
This commit is contained in:
parent
06722c4871
commit
cf5bcd8020
31
arrays_and_strings/matrix-sum-diagonals.py
Normal file
31
arrays_and_strings/matrix-sum-diagonals.py
Normal file
@ -0,0 +1,31 @@
|
||||
#!/bin/python3
|
||||
# Given a square matrix, calculate the absolute difference between the sums of its diagonals.
|
||||
|
||||
import math
|
||||
import os
|
||||
import random
|
||||
import re
|
||||
import sys
|
||||
|
||||
def diagonal_difference(arr):
|
||||
|
||||
diag_1 = 0
|
||||
diag_2 = 0
|
||||
|
||||
i, j = 0, len(arr) - 1
|
||||
|
||||
while i < len(arr) and j >= 0:
|
||||
|
||||
diag_1 += arr[i][i]
|
||||
diag_2 += arr[i][j]
|
||||
i += 1
|
||||
j -= 1
|
||||
|
||||
return diag_1, diag_2, abs(diag_1 - diag_2)
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
|
||||
print(diagonal_difference(arr))
|
Loading…
x
Reference in New Issue
Block a user