Advertisement
Guest User

Untitled

a guest
Dec 9th, 2018
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.52 KB | None | 0 0
  1. def sum_prod_diags(matrix):
  2.    
  3.     n = len(matrix)
  4.     psum =[1] * (2*n-1)
  5.     pdif = {}
  6.    
  7.    
  8.     n = len(matrix)
  9.     for i in range(0, n ):
  10.         for j in range(0, len(matrix[i])):
  11.             dif = i-j
  12.             if (dif in pdif):
  13.                 pdif[dif] *= matrix[i][j]
  14.             else:
  15.                 pdif[dif] = matrix[i][j]
  16.             psum[i+j] *= matrix[i][j]
  17.     a = 0
  18.     for i in range(0, len(psum)): a += psum[i]
  19.     b = 0
  20.     for key in pdif:
  21.         b += pdif[key]
  22.     return b-a
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement