Loading Now

sumMainDiagonal

Ma trận a được gọi là ma trận vuông nếu a là một ma trận có số hàng bằng số cột, các phần tử thuộc đường chéo chính là các phần tử a[0][0], a[1][1], a[2][2]...

Cho ma trận a gồm các số nguyên, Hãy tính tổng các phần tử thuộc đường chéo chính.

Ví dụ:

  • Với a = [[1,-1,2],[2,1,6],[5,-2,5]], thì sumMainDiagonal(a)=7.
    Giải thích:
    Tổng trên đường chéo chính là : 1+1+5=7.
  • Với a = [[1,-8,-4,3],[8,7,-3,1],[1,-2,-3,0],[1,-1,-2,4]], thì sumMainDiagonal(a)=9.

Đầu vào/Đầu ra:

  • [Thời gian chạy] 0.5s với C++, 3s với Java và C#, 4s với Python, Go và JavaScript.

  • [Đầu vào] Matrix: Integer: a
    1 ≤ a.size() = a[i].size() ≤ 103.
    | a[i][i] | ≤ 105
  • [Đầu ra] Integer
    Tổng các phần tử trên đường chéo chính.

Post Comment

Contact