aboutsummaryrefslogtreecommitdiff
path: root/matrix_diagonal_sum.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'matrix_diagonal_sum.cpp')
-rw-r--r--matrix_diagonal_sum.cpp8
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