题目描述:
给定一个m*n的矩阵,如果有一个元素是0,就把该元素所在的行和列上的元素全置为0,要求使用原地算法。
拓展: 你的算法有使用额外的空间吗?
一种比较直接的算法是利用O(m,n)的空间,但是这不是一个好的解法
使用简单的改进可以在O(m+n)的空间解决这个问题,但是还不是最佳的解法
你能在常量级的空间复杂度内解决这个问题吗?
解题思路:
- 这道题我上来直接通过遍历的方式,每遇到一个 0,就将它的行和列就设置为 0
- 这样做当然是不对的,因为这样做的结果就是整个数组最后都变为 0
- 这道题应该首先将元素为 0 的位置给记下来,当将所有的结果都记下来之后,再依次遍历存储的结果,将对应的位置设置为 0
- 和 1) 中的相比,这里多了一个中间环节,就是先存储位置,再依次处理对应位置上的元素,这样就避免出现 1)中素有元素都改为 0 的尴尬结果
ArrayList<Integer> rows = new ArrayList<>();
ArrayList<Integer> cols = new ArrayList<>();
public void setZeroes(int[][] matrix) {
for(int i = 0; i < matrix.length; i++){
for(int j = 0; j < matrix[0].length; j++){
if(matrix[i][j] == 0){
// 存储元素的位置
rows.add(i);
cols.add(j);
}
}
}
// 依次处理对应位置上的元素
for(Integer row : rows){
for(int i = 0; i < matrix[0].length; i++){
matrix[row][i] = 0;
}
}
for(Integer col :cols){
for(int i = 0 ; i < matrix.length; i++){
matrix[i][col] = 0;
}
}
}