day3——矩阵专题

最新推荐文章于 2023-08-16 18:22:58 发布

电子厂高级技工

最新推荐文章于 2023-08-16 18:22:58 发布

阅读量93

点赞数

分类专栏：蓝桥杯备赛文章标签：算法 leetcode 职场和发展 Powered by 金山文档

本文链接：https://blog.csdn.net/xmp1600486173/article/details/128962093

版权

蓝桥杯备赛专栏收录该内容

9 篇文章 0 订阅

订阅专栏

文章介绍了三个矩阵相关的问题：计算最富有客户的资产总量，找出二进制矩阵中的特殊位置，以及如何水平翻转并反转二进制矩阵。解决方案涉及使用map函数求和，通过enumerate和zip函数遍历矩阵，以及使用双指针和异或操作来处理矩阵变换。

摘要由CSDN通过智能技术生成

题目1

给你一个 m x n 的整数网格 accounts ，其中 accounts[i][j] 是第 i 位客户在第 j 家银行托管的资产数量。返回最富有客户所拥有的资产总量。

客户的资产总量就是他们在各家银行托管的资产数量之和。最富有客户就是资产总量最大的客户。

示例 1：

输入：accounts = [[1,2,3],[3,2,1]]
输出：6
解释：第 1 位客户的资产总量 = 1 + 2 + 3 = 6
第 2 位客户的资产总量 = 3 + 2 + 1 = 6
两位客户都是最富有的，资产总量都是 6 ，所以返回 6 。

代码

class Solution:
    def maximumWealth(self, accounts: List[List[int]]) -> int:
        return max(map(sum,accounts))

难点：map函数，map（a，b）对a中的元素进行b规则，实现元素映射

题目2——二进制矩阵中的特殊位置

给你一个大小为 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

思路：题目给定的一个m×n 的矩阵mat，并且矩阵中的任意元素都是1或0，现在根据特殊位置的定义，我要枚举每一个位置，然后判断每个位置是否符合要求，因为我们的矩阵中的每一个元素只能是1或0，因此我们可以通过预处理的方式来快速得到每一行和列中1的个数

代码：

class Solution:
    def numSpecial(self, mat: List[List[int]]) -> int:
#行元素之和
        rows_sum = [sum(row) for row in mat]
#列元素之和
        cols_sum = [sum(col) for col in zip(*mat)]
        res = 0
#遍历mat中的每一个元素以及每个元素对应的索引，i和j是索引，row和x是对应列表中的元素
        for i ,row in enumerate(mat):
            for j ,x in enumerate(row):
                if x==1 and rows_sum[i] == 1 and cols_sum[j] == 1:
                    res += 1
        return res

难点：

zip函数，实现以列为单位的列表

enumerate函数：将可遍历的数据对象（如列表、元组、字符串）组合成一个索引序列，同时列出数据和对应的数据下标

题目3

给定一个 n x n 的二进制矩阵 image ，先水平翻转图像，然后反转图像并返回结果。

水平翻转图片就是将图片的每一行都进行翻转，即逆序。

例如，水平翻转 [1,1,0] 的结果是 [0,1,1]。

反转图片的意思是图片中的 0 全部被 1 替换， 1 全部被 0 替换。

例如，反转 [0,1,1] 的结果是 [1,0,0]。

代码

class Solution:
    def flipAndInvertImage(self, A: list[list[int]]) -> list[list[int]]:
        for row in A:
            for k,_ in enumerate(row):row[k] = 1-row[k]
            i,j = 0,len(row)-1
            while i<j:
                row[i], row[j] = row[j],row[i]
                i,j = i+1,j-1
        return A

思路：

遍历两遍，

对原列表做两次翻转工作

首尾反转采用双指针，一个指向头，一个指向尾，两个位置交换并且指针逐渐往中间靠

返回原列表

思路2：

见缝插针法

首先我们一行的第一个数，找到它对应的数，也就是这一行最后一个

如果这两个数是不同的，比如说一个是 1，一个是 0，那么先 10 反转，则一个是 0，一个是 1，再左右翻转，又变回一个是 1，一个是 0

这说明当两个数是不同的时候，不用做任何事情

当两个数相同的时候，要同时异或或被 1 减，即 10 反转

注意，循环的范围应该是 range(len(row) + 1) // 2)，不能忘了加一。因为，如果列数为奇数，那么中间的数虽然不要左右交换，但是10还是要反转的，因此要多一次循环，相同于中间的数与自己是相同的，要反转。

代码：

class Solution:
    def flipAndInvertImage(self, A: List[List[int]]) -> List[List[int]]:
        for row in A:
            for j in range ((len(row)+1)//2):
                if row[j] ==row [-1-j]:
                    row[j] = row[-1-j] = 1-row[j]
        return A