Sum of the Diagonals

Carvia Tech | June 21, 2019 | 1 min read | 5 views


Given a square matrix of size N by N, calculate the sums of its diagonals. There are two diagonals in each array.

def get_sum_of_diagonals(mat):
    sd1 = 0
    sd2 = 0
    for i in range(0, len(mat)):
        for j in range(0, len(mat)):
            if (i == j):
                sd1 += mat[i][j]
            if ((i + j) == (len(mat)-1)):
                sd2 += mat[i][j]
    return sd1+sd2

mat = [[1,2,9],[4,5,6],[18,8,12]]
print(get_sum_of_diagonals(mat))
Output
50

Top articles in this category:
  1. Python coding challenges for interviews
  2. Google Data Scientist interview questions with answers
  3. Top 100 interview questions on Data Science & Machine Learning
  4. Python Flask Interview Questions
  5. Sequence of Differences in Python
  6. Pass the ball game: NxN matrix in python
  7. Derivative of 1/x & Tossing a coin problem


Find more on this topic:
Machine Learning image
Machine Learning

Data science, machine learning, python, R, big data, spark, the Jupyter notebook, and much more

Last updated 1 week ago


Recommended books for interview preparation:

This website uses cookies to ensure you get the best experience on our website. more info