清除行列

题目描述

请编写一个算法，若N阶方阵中某个元素为0，则将其所在的行与列清零。

给定一个N阶方阵int[][](C++中为vector<vector>)mat和矩阵的阶数n，请返回完成操作后的int[][]方阵(C++中为vector<vector>)，保证n小于等于300，矩阵中的元素为int范围内。

测试样例：

[[1,2,3],[0,1,2],[0,0,1]], 3

返回：[[0,0,3],[0,0,0],[0,0,0]]

--------------------------------------------

思路：由于当前值若为0，直接处理会影响后面行列的判断，所以不能直接一遍处理，要先扫描一遍记录为零的行列;

再次扫描，处理0.。

C/C++ (clang++ 3.3)

1

class Clearer {

2

public:

3

    vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {

4

        if (n==0)

5

            return mat;

6

        int r_flag[300] = {0};//记录应该为0的行

7

        int c_flag[300] = {0};//记录应该为0的列

8

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

9

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

10

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

11

                    //mat[i].assign(n, 0);

12

                    c_flag[j] = 1;

13

                    r_flag[i] = 1;   

14

                }

15

            }

16

        }

17

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

18

            if (r_flag[i]==1){

19

                mat[i].assign(n, 0);

20

            }

21

            if (c_flag[i]==1){

22

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

23

                    mat[j][i] = 0;

24

            }

25

        }

26

27

        return mat;

28

    }

29

};

您的代码已保存