零矩阵

/*
编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。
示例 1：
输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]
示例 2：
输入：
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出：
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]
来源：力扣（LeetCode）
*/
/*
说明：
1、第一行保存每列是否需要清零，第一行遍历时标记，最后判断第一行是否需要清零；
2、最后一行时去清零每一列，因此最后一行跟第一行一样用标记，最后判断是否要清零这一行；
3、非第一行和最后一行，则清零整行；
4、最后判断第一行和最后一行是否需要全部清零；
*/

#include <stdio.h>
#include <stdbool.h>

void setZeroes(int** matrix, int matrixSize, int* matrixColSize){
	int i, j, k;
	int row = matrixSize;
	int col = *matrixColSize;
	bool firstRow = false;
	bool lastRow = false;
	
	for (i = 0; i < row; i++) {
		for (j = 0; j < col; j++) {
			//标记第一行是否清零
			if (i == 0 && matrix[i][j] == 0) {
				firstRow = true;
				j = col; //跳到下一行
				continue;
			} 
			
			//最后一行，清理所有需要清理的列，如果这个元素为0，则行也要清零
			if (i + 1 == row && (matrix[i][j] == 0 || matrix[0][j] == 0)) {
				if (matrix[i][j] == 0) {
					lastRow = true;
					matrix[0][j] = 0;
				}
				//清零该列
				for (k = 1; k <= i; k++) {
					matrix[k][j] = 0;
				}
				continue;
			}
			
			//非第一行和最后一行，该元素为0，只清零该行，因为最后一行前面已经过滤处理了，肯定不是最后一行
			if (i != 0 && i + 1 < row && matrix[i][j] == 0) {
				//清零该行
				for (k = 0; k < col; k++) {
					if(k > j && matrix[i][k] == 0) {
						matrix[0][k] = 0;
						continue;
					}
					matrix[i][k] = 0;
				}
				matrix[0][j] = 0; //列标记到第一行
				j = col; //直接跳到下一行
				continue;
			}
		}
	}
	
	//如果第一行有0，则最后把第一行清了
	if (firstRow == true) {
		for (k = 0; k < col; k++) {
			matrix[0][k] = 0;
		}
	}
	if (lastRow == true) {
		for (k = 0; k < col; k++) {
			matrix[row - 1][k] = 0;
		}
	}
}