面试题 01.08题 零矩阵(C++)- pair

题目链接:https://leetcode-cn.com/problems/zero-matrix-lcci/
题目如下:
在这里插入图片描述
在这里插入图片描述

class Solution {
    
    
public:
    void setZeroes(vector<vector<int>>& matrix) {
    
    
        if(matrix.size()==0) return ;//特判

        pair<int,int> p[matrix.size()*matrix[0].size()];
        int cnt=0;

        //记录下所有值为0的行列下标
        for(int i=0;i<matrix.size();i++){
    
    
            for(int j=0;j<matrix[0].size();j++){
    
    
                if(matrix[i][j]==0) p[cnt++]=make_pair(i,j);
            }
        }

        for(int i=0;i<cnt;i++){
    
    
            //对每个行和列清零
            for(int j=0;j<matrix.size();j++)//行
            matrix[j][p[i].second]=0;

            for(int j=0;j<matrix[0].size();j++)//列
            matrix[p[i].first][j]=0;
        }
    }
};

猜你喜欢

转载自blog.csdn.net/qq_40467670/article/details/121447651