Java - 如果二维数组包含目标数字，则将其行/列清零

Question

给定一个看起来像这样的 mxn 矩阵：
1034
1234
3560

需要输出如下内容：
0000
1030
0000

*目标数为0。

这是我的解决方案，但我认为它不是很有效（空间和运行时间我认为是 O(m^2 * n)）并且想知道是否有更简单和更有效的方法来做到这一点。如果有，是什么？

int[][] m = { { 1, 0, 3, 4 }, { 1, 2, 3, 4 }, { 3, 5, 6, 0 } };  
m = zero(m, 0);

public static int[][] zero(int[][] m, int num) {

    int rows = m.length;
    int columns = m[0].length;
    int [][] myInt = new int[rows][];
    for(int i = 0; i < rows; i++){
        myInt[i] = m[i].clone();
    }
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < columns; j++) {
            if (myInt[i][j] == num) {
                m[i] = new int[columns];
                for(int k = 0; k < rows; k++){
                    m[k][j] = 0;
                }
                break;
            }
        }
    }       
    return m;
}

基本上我先克隆输入矩阵，然后遍历每一行并检查该行是否包含我的目标数字。如果是，那么我将原始矩阵中的整行设置为零。然后我执行另一个循环以将包含目标编号的列设置为零。我在一开始就克隆了矩阵，因此检查总是针对克隆的参考矩阵，而不是每次迭代时修改的参考矩阵。

Answer 1

我刚刚遇到了这个问题，并为此制定了解决方案。

我希望就代码更好/更差以及运行时和空间复杂度方面的代码提供一些反馈。这一切都很新:)

public static void zeroMatrix(int[][] arr1)
{
    ArrayList<Integer> coord = new ArrayList<>();

    int row = arr1.length; 
    int column = arr1[0].length;
    for(int i=0; i < row; i++)
    {
        for(int j=0; j < column; j++)
        {
            if(arr1[i][j]==0)
            { 
                coord.add((10*i) + j);
            }       
        }
    }

    for(int n : coord)
    {
         int j=n%10;
         int i=n/10; int k=0;
        int l=0;
        while(k<row)
        {
            arr1[k][j]=0;
            k++;
        }
        while(l<column)
        {
            arr1[i][l]=0;
            l++;
        }
    }
}

Answer 2

我建议将 BitSet 用于行/列索引。

public static void zero(int[][] m, int num) {

    int rows = m.length;
    int columns = m[0].length;
    BitSet rowsToClear = new BitSet(rows);
    BitSet columnsToClear = new BitSet(columns);
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < columns; j++) {
            if (m[i][j] == num) {
                rowsToClear.set(i);
                columnsToClear.set(j);
            }
        }
    }
    for (int i = rowsToClear.nextSetBit(0); i >= 0;
             i = rowsToClear.nextSetBit(i + 1)) {
        Arrays.fill(m[i], 0);
    }
    for (int j = columnsToClear.nextSetBit(0); j >= 0;
            j = columnsToClear.nextSetBit(j + 1)) {
        for (int i = 0; i < rows; ++i) {
            m[i][j] = 0;
        }
    }
    //return m;
}