Question

How to find the sum of each row and each column of a matrix (2D array) in Rust

Answer 1

fn main() {
    let arr: [[i32; 4]; 3] = [
        [1, 2, 3, 5],
        [4, 5, 6, 5],
        [7, 8, 9, 5],
    ];

    let rows = arr.len();
    let cols = arr[0].len();

    for i in 0..rows {
        let sum_row: i32 = arr[i].iter().sum();
        println!("Sum of row: {} = {}", i, sum_row);
    }

    for i in 0..cols {
        let sum_col: i32 = (0..rows).map(|j| arr[j][i]).sum();
        println!("Sum of col: {} = {}", i, sum_col);
    }
}




/*
run:
  
Sum of row: 0 = 11
Sum of row: 1 = 20
Sum of row: 2 = 29
Sum of col: 0 = 12
Sum of col: 1 = 15
Sum of col: 2 = 18
Sum of col: 3 = 15
  
*/

 

70+ SQL courses for beginners and professionals