【数组】1582. 二进制矩阵中的特殊位置（简单）

【题目】
给你一个大小为 rows x cols 的矩阵 mat，其中 mat[i][j] 是 0 或 1，请返回 矩阵 mat 中特殊位置的数目 。特殊位置 定义：如果 mat[i][j] == 1 并且第 i 行和第 j 列中的所有其他元素均为 0（行和列的下标均 从 0 开始 ），则位置 (i, j) 被称为特殊位置。
【示例1】
输入：mat = [[1,0,0],
[0,0,1],
[1,0,0]]
输出：1
解释：(1,2) 是一个特殊位置，因为 mat[1][2] == 1 且所处的行和列上所有其他元素都是 0
【示例2】
输入：mat = [[1,0,0],
[0,1,0],
[0,0,1]]
输出：3
解释：(0,0), (1,1) 和 (2,2) 都是特殊位置
【代码】

class Solution {
public:
    int numSpecial(vector<vector<int>>& mat) {
        vector<int> row;
        vector<int> col;
        int cnt=0;
        for(int i=0;i<mat.size();i++){
            int base=0;
            for(int j=0;j<mat[0].size();j++)
                base+=mat[i][j];            
            row.push_back(base);
        }
        for(int i=0;i<mat[0].size();i++){
            int base=0;
            for(int j=0;j<mat.size();j++)
                base+=mat[j][i];
            col.push_back(base);
        }
        for(int i=0;i<mat.size();i++)
            for(int j=0;j<mat[0].size();j++)
                if(mat[i][j]&&row[i]==1&&col[j]==1)
                    cnt++;
        return cnt;
    }
};