Cracking the coding interview--Q1.7

原文

Write an algorithm such that if an element in an MxN matrix is 0, its entire row and column are set to 0.

译文

写一个算法，如果一个MxN中的一个元素为0，那么将这个元素所在的行和列都设置为0。

解答

遍历一次矩阵，当遇到元素等于0时，记录下这个元素对应的行和列。并设置一个标记。在遍历完之后，再根据之前的标记设置矩阵的值。

import java.util.BitSet;



public class Main {



    public static void setZero(int[][] mat, int m, int n) {

        BitSet lineFlag = new BitSet();

        BitSet rowFlag = new BitSet();

        for(int i = 0; i < m; i++) {

            for(int j = 0; j < n; j++) {

                if(mat[i][j] == 0) {

                    lineFlag.set(i);

                    rowFlag.set(j);

                }

            }

        }

        for(int i = 0; i < m; i++) {

            for(int j = 0; j < n; j++) {

                if(lineFlag.get(i) || rowFlag.get(j)) {

                    mat[i][j] = 0;

                }

            }

        }

    }



    public static void main(String args[]) {

        int a[][] = { { 1, 2, 3, 4 }, { 5, 6, 0, 8 }, { 9, 0, 11, 12 },

                { 13, 14, 15, 16 } };

        for (int i = 0; i < 4; ++i) {

            for (int j = 0; j < 4; ++j)

                System.out.print(a[i][j] + " ");

            System.out.println();

        }

        setZero(a, 4, 4);

        System.out.println("------------------");

        for (int i = 0; i < 4; ++i) {

            for (int j = 0; j < 4; ++j)

                System.out.print(a[i][j] + " ");

            System.out.println();

        }

    }

}