《程序员面试金典》(第六版)习题:仅为记录一下以加强印象,不为商业用途,如有侵权请联系删除。以下源码和解释参考了书中源码以及这里:https://github.com/alexhagiopol/cracking-the-coding-interview
//C++ CODE//
void setZero(vector<vector<int>> & inputMatrix)
{
int rowNum = inputMatrix.size();
int columnNum = inputMatrix[0].size();
bool firstZeroRow = false;
bool firstZeroCol = false;
for (int i = 0; i < columnNum; i++)
{
if (inputMatrix[0][i] == 0)
{
firstZeroRow = true;
break;
}
}
for (int j = 0; j < rowNum; j++)
{
if (inputMatrix[j][0] == 0)
{
firstZeroCol = true;
break;
}
}
for (int i = 1; i < rowNum; i++)
{
for (int j = 1; j < columnNum; j++)
{
if (inputMatrix[i][j] == 0)
{
inputMatrix[0][j] = 0;
inputMatrix[i][0] = 0;
}
}
}
for (int i = 1; i < rowNum; i++)
{
if (inputMatrix[i][0] == 0)
{
for (int j = 1; j < columnNum; j++)
{
inputMatrix[i][j] = 0;
}
}
}
for (int i = 1; i < columnNum; i++)
{
if (inputMatrix[0][i] == 0)
{
for (int j = 1; j < rowNum; j++)
{
inputMatrix[j][i] = 0;
}
}
}
if (firstZeroRow)
{
for (int i = 0; < columnNum; i++)
{
inputMatrix[0][i] = 0;
}
}
if (firstZeroCol)
{
for (int j = 0; j < rowNum; j++)
{
inputMatrix[j][0] = 0;
}
}
}
可以遍历矩阵inputMatrix
[
M
]
[
N
]
[M][N]
[M][N]的每一个元素,如果该元素为零则将相应的行和列的元素全部赋值为零。这样的做法看起来可行,其实是一个陷阱。假如矩阵只有元素inputMatrix
[
0
]
[
0
]
[0][0]
[0][0]为零,正确的答案是将该矩阵第零行和第零列的所有元素赋值为零即可。但是当将该矩阵第零行和第零列的所有元素赋值为零后,之后再遍历到第零行和第零列的元素之后就会将整个矩阵的元素赋值为零,因为之前的操作已经将该矩阵第零行和第零列的所有元素赋值为零。
可行的做法是建立两个额外的数组,
R
[
M
]
R[M]
R[M]和
L
[
N
]
L[N]
L[N],一个大小为矩阵行元素个数的大小,用来记录包含零元素的行。一个大小为矩阵列元素个数的大小,用来记录包含零元素的列。遍历矩阵的所有元素inputMatrix
[
i
]
[
j
]
[i][j]
[i][j],当该元素为零时,将数组元素
R
[
i
]
R[i]
R[i]和
L
[
j
]
L[j]
L[j]进行标记来表明相应的行和列存在零元素。最后遍历这两个数组,当索引对应的元素被标记时,就将相应的行和列的所有元素清零。算法结束。但是该算法用到了额外的空间,算法的空间复杂度为O(M+N)。
为了进一步优化算法,将算法的空间复杂度为优化为O(1)。改进算法首先判断第一行和第一列是否有零元素存在。之后遍历除第一行和第一列之外的所有元素。如果该元素inputMatrix
[
i
]
[
j
]
[i][j]
[i][j]为零则将inputMatrix
[
0
]
[
j
]
[0][j]
[0][j]和inputMatrix
[
i
]
[
0
]
[i][0]
[i][0]进行标记。改进算法与前一步的算法类似,只不过将第一行和第一列作为额外的数组来进行标记零元素,并首先判断第一行和第一列是否有零元素。最后根据第一行的每一个元素是否为零来对相应的列清零,根据第一列的每一个元素是否为零来对相应的行清零,这里的操作不会涉及第零行和第零列。根据第一行是否有零元素,对第一行来进行清零,根据第一列是否有零元素,对第一列来进行清零。
改进算法的时间复杂度为O(M+N),空间复杂度为优化为O(1)。