aboutsummaryrefslogtreecommitdiff
path: root/set_matrix_zeroes.c
blob: be0e3328dfd6f37580093b8b5ad3953f7bc820af (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
// 73. Set Matrix Zeroes
#include "leetcode.h"

/*
 * given an 'm x n' integer 'matrix', if an element is '0', set its entire row
 * and columns to '0's. you must do it in place
 */

void setZeroes(int **matrix, int matrixSize, int *matrixColSize) {
  bool *r = (bool *)calloc(matrixSize, sizeof(bool));
  bool *c = (bool *)calloc(*matrixColSize, sizeof(bool));
  for (int i = 0; i < matrixSize; ++i)
    for (int j = 0; j < *matrixColSize; ++j)
      if (!matrix[i][j]) {
        r[i] = true;
        r[j] = true;
      }
  for (int i = 0; i < matrixSize; ++i)
    for (int j = 0; j < *matrixColSize; ++j)
      if (r[i] || c[j])
        matrix[i][j] = 0;
}

int main() {
  int m1i[3][3] = {{1, 1, 1}, {1, 0, 1}, {1, 1, 1}},
      m2i[3][4] = {{0, 1, 2, 0}, {3, 4, 5, 2}, {1, 3, 1, 5}};
  struct two_d_arr m1, m2;
  two_d_arr_init(&m1, ARRAY_SIZE(m1i), ARRAY_SIZE(m1i[0]), m1i);
  two_d_arr_init(&m2, ARRAY_SIZE(m2i), ARRAY_SIZE(m2i[0]), m2i);
  setZeroes(m1.arr, m1.row_size, m1.col_size);
  setZeroes(m2.arr, m2.row_size, m2.col_size);
  for (int i = 0; i < m1.row_size; i++) {
    for (int j = 0; j < m1.col_size[i]; j++)
      printf("%d ", m1i[i][j]); // expect: [[1,0,1],[0,0,0],[1,0,1]]
    printf("\n");
  }
  for (int i = 0; i < m2.row_size; i++) {
    for (int j = 0; j < m2.col_size[i]; j++)
      printf("%d ", m2i[i][j]); // expect: [[0,0,0,0],[0,4,5,0],[0,3,1,0]]
    printf("\n");
  }
  two_d_arr_free(&m1);
  two_d_arr_free(&m2);
}