题目:
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 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] ]
进阶:
- 一个直接的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
- 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
- 你能想出一个常数空间的解决方案吗?
思路:我自己写的代码不对,然后就在网上找了大神的代码。原文链接:https://blog.csdn.net/derrantcm/article/details/47371979。大神代码的运行速度是最快的,但是和leetcode中最快的代码好像有点不一样。
这个题目中要求在原地完成,并且希望在常数空间内完成,这个要求我有点不太懂。以下代码的基本思想是首先对数组中的0元素进行标记。以第一行和第一列作为一个标志位,若某个元素为0,那么在它对应的第一行和第一列的位置上都设置为0;同时为了区分这个0元素是否为与第一行和第一列上,还设置了两个标志位。在标记完成数组之后,分两部分进行置零操作,一部分是第一行和第一列中的元素,这些元素是否全部置零是根据之前的两个标志位来判断;第二部分是剩余部分,它们是否置零是由对应的第一行或者第一列中是否为零来判断。这种思路应该就是使用了常数空间,只使用了额外的两个标志位。
代码:
public class Solution {
public void setZeroes(int[][] matrix) {
// 第一行被设置的标志 这个标志位标记的是0元素是否在第一行
boolean rowFlag = false;
// 第一列被设置的标志 标记这个0元素是否在第一列
boolean colFlag = false;
for (int i = 0; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
if (matrix[i][j] == 0) {
// 标记第一行要被设置
if (i == 0)
{
rowFlag = true;
}
// 标记第一列要被设置
if (j == 0){
colFlag = true;
}
// 在行列在标记要设置为0的行和列
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
// 对第二行第二列开始的元素设置0
for (int i = 1; i < matrix.length; i++) {
for (int j = 1; j < matrix[0].length; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0) {
matrix[i][j] = 0;
}
}
}
// 设置第一行为0
if (rowFlag) {
for (int j = 0; j < matrix[0].length; j++) {
matrix[0][j] = 0;
}
}
// 设置第一列为0
if (colFlag) {
for (int i = 0; i < matrix.length; i++) {
matrix[i][0] = 0;
}
}
}
}
自己写的代码
我还不知道是哪里错了。。。。。。。应该不是使用常数的空间,并且好像参数的传递也有点问题吧
class Solution {
public void setZeroes(int[][] matrix) {
int m=matrix.length;
int n=matrix[0].length;
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(matrix[i][j]==0)
ss(matrix,i,j);
continue;
}
}
}
private void ss(int[][]a,int b,intc){ //总是提示这一行不正确
for(int i=0;i<n;i++){
a[b][i]==0;
}
for(int j=0;j<m;j++){
a[j][c]==0;
}
}
}
以下是leetcode中执行最快的代码:
基本思想和上面的分析一致,只不过只使用另一个标志位。
class Solution {
public void setZeroes(int[][] matrix) {
int col0 = 1;
int m = matrix.length;
int n = matrix[0].length;
for(int i = 0; i < m; i++){
if(matrix[i][0] == 0)
col0 = 0;
for(int j = 1; j < n; j++)
if(matrix[i][j] == 0)
matrix[0][j] = matrix[i][0] = 0;
}
for(int i = m-1; i >= 0; i--){
for(int j = n-1; j >= 1; j--)
if( matrix[0][j] == 0 || matrix[i][0] == 0)
matrix[i][j] = 0;
if(col0 == 0)
matrix[i][0] = 0;
}
}
}