1、思路

• 首先要遍历第一行与第一列，确定其中是否有0元素，做好标记；

• 用第一行与第一列来做标记，遍历数组，若数组中存在一个0元素，那么把该元素所在的第一行和第一列都置为0；

• 再次遍历数组，判断当前元素的行首或列首是否为0，若是，则将当前元素置0；

• 最后再判断是否需要将第一行与第一列置0。

2、代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        if (matrix.size() == 0) return;
        bool firstRowHasZero = false, firstColHasZero = false;  //标记第一行与第一列是否存在0

        for (int i = 0; i < matrix.size(); ++ i)    //遍历第一行找是否有0
        {
            if (matrix[i][0] == 0)
            {
                firstColHasZero = true;
                break;
            }
        }

        for (int j = 0; j < matrix[0].size(); ++ j) //遍历第一列找是否有0
        {
            if (matrix[0][j] == 0)
            {
                firstRowHasZero = true;
                break;
            }
        }

        //找到除第一行第一列以外的0，并将0元素的行首与列首标记为0
        for (int i = 1; i < matrix.size(); ++ i)
        {
            for (int j = 1; j < matrix[0].size(); ++ j)
            {
                if (matrix[i][j] == 0)
                {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }

        //根据行首与列首的0来对矩阵内元素置0
        for (int i = 1; i < matrix.size(); ++ i)
        {
            for (int j = 1; j < matrix[0].size(); ++ j)
            {
                if (matrix[i][0] == 0 || matrix[0][j] == 0)
                {
                    matrix[i][j] = 0;
                }
            }
        }

        if (firstColHasZero)    //判断是否将第一列置零
        {
            for (int i = 0; i < matrix.size(); ++ i)
            {
                matrix[i][0] = 0;
            }
        }

        if (firstRowHasZero)    //判断是否将第一行置零
        {
            for (int j = 0; j < matrix[0].size(); ++ j)
            {
                matrix[0][j] = 0;
            }
        }        
    }
};