From cf5bcd802008f9d0e548f533d8818a05eaebadca Mon Sep 17 00:00:00 2001 From: bt3gl <138340846+cypher-bt3gl@users.noreply.github.com> Date: Sat, 29 Jul 2023 20:24:59 -0700 Subject: [PATCH] Add files via upload --- arrays_and_strings/matrix-sum-diagonals.py | 31 ++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 arrays_and_strings/matrix-sum-diagonals.py diff --git a/arrays_and_strings/matrix-sum-diagonals.py b/arrays_and_strings/matrix-sum-diagonals.py new file mode 100644 index 0000000..6999fa4 --- /dev/null +++ b/arrays_and_strings/matrix-sum-diagonals.py @@ -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))