Count Servers that Communicate LeetCode solution in Rust

1
impl Solution {
2
  pub fn count_servers(grid: Vec<Vec<i32>>) -> i32 {
3
    let n: usize = grid.len();
4
    let m: usize = grid[0].len();
5
    let mut row: Vec<i32> = vec![0; n];
6
    let mut col: Vec<i32> = vec![0; m];
7

8
    let mut res: i32 = 0;
9

10
    //count number of servers in each row and column
11
    for i in 0..n {
12
      for j in 0..m {
13
        if grid[i][j] == 1 {
14
          row[i] += 1;
15
          col[j] += 1;
16
        }
17
      }
18
    }
19

20
    //count number of servers not alone in either row or column
21
    for i in 0..n {
22
      for j in 0..m {
23
        if grid[i][j] == 1 && (row[i] > 1 || col[j] > 1) {
24
          res += 1;
25
        }
26
      }
27
    }
28
    return res;
29
  }
30
}