mirror of
https://github.com/autistic-symposium/master-algorithms-py.git
synced 2025-05-22 16:31:15 -04:00
Add files via upload
This commit is contained in:
parent
06722c4871
commit
cf5bcd8020
1 changed files with 31 additions and 0 deletions
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…
Add table
Add a link
Reference in a new issue