做算法想要提高,就要多思考,多写代码,多做,并且看到题时先不要看别人的解析,先自己琢磨琢磨,然后再看别人的解析,这样会比较好的锻炼自己的思维能力。
题目:
给定一个 m x n
的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
算法思路:如果检查到有元素0,我们需要把这个元素所在的行和列都变成0,所以我们保存行和列的信息,这里我们使用两个bool类型的数组进行保存,也就是把0所在的行和列都为true,之后再次遍历时为true的变为0,其他的数,遍历时直接跳过,时间复杂度还算可以,因为创建了两个bool数组,所以空间复杂度可能会稍微高一些。
具体代码:
public class Solution { public void SetZeroes(int[][] matrix) { int m = matrix.Length; int n = matrix[0].Length; //设置两个标记数组 //标记行 bool[] row = new bool[m]; //标记列 bool[] col = new bool[n]; //循环遍历矩阵 记录0出现的位置 //用标记数组进行标记 for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (matrix[i][j]==0) { row[i] = col[j] = true; } } } for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (row[i]||col[j]) { matrix[i][j] = 0; } } } } }
希望能和大家一起进步,加油,打工人