[
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
]

trace([[1, 4], [4, 1]]) ➞ 2

trace([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) ➞ 15

trace([[1, 0, 1, 0], [0, 2, 0, 2], [3, 0, 3, 0], [0, 4, 0, 4]]) ➞ 10

self.assertEqual(trace([[1, 4], [4, 1]]), 2)
self.assertEqual(trace([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), 15)
self.assertEqual(trace([[1, 0, 1, 0], [0, 2, 0, 2], [3, 0, 3, 0], [0, 4, 0, 4]]), 10)

پاسخ های کاربران به این تمرین