diff options
Diffstat (limited to 'matrix-diagonal-sum/src/main.rs')
-rw-r--r-- | matrix-diagonal-sum/src/main.rs | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/matrix-diagonal-sum/src/main.rs b/matrix-diagonal-sum/src/main.rs new file mode 100644 index 0000000..8ee95a0 --- /dev/null +++ b/matrix-diagonal-sum/src/main.rs @@ -0,0 +1,11 @@ +fn main() { + println!("Hello, world!"); +} + +struct Solution {} +impl Solution { + pub fn diagonal_sum(mat: Vec<Vec<i32>>) -> i32 { + mat.iter().enumerate() + .fold(0, |acc, (i, row)| acc + row[i] + if i != row.len()-1 -i {row[row.len()-1 -i]} else {0}) + } +} |