[英雄星球六月集训LeetCode解题日报] 第11日矩阵

七水shuliang

已于 2022-06-11 21:41:46 修改

阅读量113

点赞数

分类专栏：今天开刷leetcode 英雄星球六月集训文章标签： leetcode 矩阵算法

于 2022-06-11 21:29:43 首次发布

本文链接：https://blog.csdn.net/liuliangcan/article/details/125238898

版权

今天开刷leetcode 同时被 2 个专栏收录

139 篇文章 2 订阅

订阅专栏

英雄星球六月集训

37 篇文章 1 订阅

订阅专栏

[英雄星球六月集训LeetCode解题日报] 第11日矩阵

一、面试题 01.08. 零矩阵

链接: 面试题 01.08. 零矩阵

1. 题目描述

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

2. 思路分析

遍历第一次标记每一行，每一列是否有0。
遍历第二次，根据标记置零。
这里用首行首列做标记。

3. 代码实现

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        用第一行来记录每列有没有0
        用第一列来记录每行有没有0
        如：
            如果(i,j)==0,那记录(0,j),(i,0)都为0，
            最后处理时，发现(0,j)为0，要么是这列有0，要么本身是0，都应该让这列变成0
        """

        m,n = len(matrix),len(matrix[0])
        row0has0,col0has0 = False,False
        for i in range(m):
            if matrix[i][0] == 0:
                col0has0 = True
                break
        for i in range(n):
            if matrix[0][i] == 0:
                row0has0 = True
                break
        
        for i in range(1,m):
            for j in range(1,n):
                if matrix[i][j] == 0:
                    matrix[0][j] = 0
                    matrix[i][0] = 0
        for i in range(1,m):
            for j in range(1,n):
                if matrix[0][j] == 0 or matrix[i][0]==0:
                    matrix[i][j] = 0

        if row0has0 :
            for i in range(n):
                matrix[0][i] = 0

        if col0has0:
            for i in range(m):
                matrix[i][0] = 0
        return matrix

二、 1848. 73. 矩阵置零

链接: 73. 矩阵置零

1. 题目描述

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用原地算法。

2. 思路分析

同上题

3. 代码实现

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        用第一行来记录每列有没有0
        用第一列来记录每行有没有0
        如：
            如果(i,j)==0,那记录(0,j),(i,0)都为0，
            最后处理时，发现(0,j)为0，要么是这列有0，要么本身是0，都应该让这列变成0
        """

        m,n = len(matrix),len(matrix[0])
        row0has0,col0has0 = False,False
        for i in range(m):
            if matrix[i][0] == 0:
                col0has0 = True
                break
        for i in range(n):
            if matrix[0][i] == 0:
                row0has0 = True
                break
        
        for i in range(1,m):
            for j in range(1,n):
                if matrix[i][j] == 0:
                    matrix[0][j] = 0
                    matrix[i][0] = 0
        for i in range(1,m):
            for j in range(1,n):
                if matrix[0][j] == 0 or matrix[i][0]==0:
                    matrix[i][j] = 0

        if row0has0 :
            for i in range(n):
                matrix[0][i] = 0

        if col0has0:
            for i in range(m):
                matrix[i][0] = 0
        return matrix

三、 1727. 重新排列后的最大子矩阵

链接: 1727. 重新排列后的最大子矩阵

1. 题目描述

给你一个二进制矩阵 matrix ，它的大小为 m x n ，你可以将 matrix 中的列按任意顺序重新排列。

请你返回最优方案下将 matrix 重新排列后，全是 1 的子矩阵面积。

在这里插入图片描述

2. 思路分析

求子矩阵面积,一般可以枚举每一行，行内枚举每一列高度,然后划定以这个高度为高的矩阵的宽度范围,单调栈求左右边界.
这题由于列可以移动,因此我们给每列高度排序,枚举时宽度设定为向大延伸到头,即可。
对每行的枚举相当于求柱状图的最大矩形面积。
实现要以dp的思想计算当前行的高度。
行内剪枝：如果从大到小排序，后边的高度乘最大宽度也到不了当前最大面积，那就不用继续算了。
实测这个剪枝意义不大，可能因为主要复杂度还是在O(n×mlogm)

3. 代码实现

class Solution:
    def largestSubmatrix(self, matrix: List[List[int]]) -> int:
        m,n = len(matrix),len(matrix[0])
        ans = 0
        for i in range(m):
            for j in range(n):
                if i > 0 and matrix[i][j] == 1:
                    matrix[i][j] = matrix[i-1][j] + 1            
            for j,h in  enumerate(sorted(matrix[i],reverse=True)):
                
                if ans >= h*n:  # 即使后边都是h，数据也不会更新，那就不用继续了。
                    break 
                ans = max(ans,h*(j+1))        
        return ans

四、1034. 边界着色

链接: 1034. 边界着色

1. 题目描述

给你一个大小为 m x n 的整数矩阵 grid ，表示一个网格。另给你三个整数 row、col 和 color 。网格中的每个值表示该位置处的网格块的颜色。

两个网格块属于同一连通分量需满足下述全部条件：

两个网格块颜色相同
在上、下、左、右任意一个方向上相邻
连通分量的边界是指连通分量中满足下述条件之一的所有网格块：

在上、下、左、右任意一个方向上与不属于同一连通分量的网格块相邻
在网格的边界上（第一行/列或最后一行/列）
请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的连通分量的边界进行着色，并返回最终的网格 grid 。

2. 思路分析

题目很长，实际题意很简单：就是求给定点能走到的所有位置，判断是否是边界。

直接BFS搜索即可。
出队处理时加个边界判断。
不能同时修改，因此用额外的数组储存要修改的位置，最后再改写。

3. 代码实现

class Solution:
    def colorBorder(self, grid: List[List[int]], row: int, col: int, color: int) -> List[List[int]]:
        m,n = len(grid),len(grid[0])
        lc = grid[row][col]
        def is_lc(x,y):  # 同色
            return grid[x][y] == lc
        def is_inbound(x,y):  # 界内
            return 0<=x<m and 0<=y<n
        def is_bianjie(x,y):  # 判这个点是不是边界
            flag = False
            return any(not is_inbound(x1,y1) or not is_lc(x1,y1) for x1,y1 in [(x,y+1),(x,y-1),(x+1,y),(x-1,y)])

        q = deque([(row,col)])
        visited = set([(row,col)])
        ans = []
        while q:
            x,y = q.popleft()
            if is_bianjie(x,y):
                ans.append((x,y))
            
            for x1,y1 in [(x,y+1),(x,y-1),(x+1,y),(x-1,y)]:
                if is_inbound(x1,y1) and is_lc(x1,y1) and (x1,y1) not in visited:
                    q.append((x1,y1))
                    visited.add((x1,y1))
        
        for x,y in ans:
            grid[x][y] = color
        return grid