diff options
Diffstat (limited to 'matrix_diagonal_sum.cpp')
-rw-r--r-- | matrix_diagonal_sum.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/matrix_diagonal_sum.cpp b/matrix_diagonal_sum.cpp index d5234b5..90e7fd5 100644 --- a/matrix_diagonal_sum.cpp +++ b/matrix_diagonal_sum.cpp @@ -10,7 +10,7 @@ class Solution { public: - int diagonalSum(vvd(int) & mat) { + int diagonalSum(vvd & mat) { int n = mat.size(), ans = 0; for (int i = 0; i < n; i++) { if (i != n - i - 1) @@ -24,9 +24,9 @@ public: int main() { Solution obj; - vvd(int) m1 = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}; - vvd(int) m2 = {{1, 1, 1, 1}, {1, 1, 1, 1}, {1, 1, 1, 1}, {1, 1, 1, 1}}; - vvd(int) m3 = {{5}}; + vvd m1 = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}; + vvd m2 = {{1, 1, 1, 1}, {1, 1, 1, 1}, {1, 1, 1, 1}, {1, 1, 1, 1}}; + vvd m3 = {{5}}; printf("%d\n", obj.diagonalSum(m1)); // expect: 25 printf("%d\n", obj.diagonalSum(m2)); // expect: 8 printf("%d\n", obj.diagonalSum(m3)); // expect: 5 |