leetcode笔记之[1582. 二进制矩阵中的特殊位置]

122 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第1天,点击查看活动详情

一、题目描述:

1582. 二进制矩阵中的特殊位置 - 力扣(LeetCode)

给你一个大小为 rows x cols 的矩阵 mat,其中 mat[i][j]01,请返回 矩阵 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) 都是特殊位置

示例 3:

输入:mat = [[0,0,0,1],
            [1,0,0,0],
            [0,1,1,0],
            [0,0,0,0]]
输出:2

示例 4:

输入:mat = [[0,0,0,0,0],
            [1,0,0,0,0],
            [0,1,0,0,0],
            [0,0,1,0,0],
            [0,0,0,1,1]]
输出:3

 

提示:

    • rows == mat.length
    • cols == mat[i].length
    • 1 <= rows, cols <= 100
    • mat[i][j] 是 0 或 1

二、思路分析:

  1. 找mat[i][j] = 1的点
  2. isValid判断i行和j列是否满足条件

注意点:

  • isValid判断时要除去mat[i][j] = 1 本身,为了快速遍历,直接将前面判断过mat[i][j] = 1的点设为0

  • 判断结束后,将mat[i][j] = 1的点复原,再从0变为1

    只要执行了isValid判断的点都需要复原

三、AC 代码:

class Solution {
    public int numSpecial(int[][] mat) {
        int rows = mat.length;
        int cols = mat[0].length;
        int count = 0;
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                if (mat[i][j] == 1) {   //找出mat[i][j] = 1 的点
                    if (isValid(mat, i, j)) {   //对满足第一个条件的点判断第二个条件是否满足
                        count++;
                    }
                    mat[i][j] = 1;  //只要执行了isValid的点都要重新复原为1
                }
            }
        }
        return count;
    }

    static boolean isValid(int[][] mat, int row, int col) {
        int rows = mat.length;
        int cols = mat[0].length;
        mat[row][col] = 0;  //将判断为1的点暂时设为0,方便遍历时判断
        for (int i = 0; i < rows; i++) {    //遍历判断(i,j)所在列的元素是否全为0,不是则返回false
            if (mat[i][col] != 0) {
                return false;
            }
        }
        for (int j = 0; j < cols; j++) {    //遍历判断(i,j)所在行的元素是否全为0,不是则返回false
            if (mat[row][j] != 0) {
                return false;
            }
        }
        return true;
    }
}