文章目录
- 1. 题目
- 2. 解题
1. 题目
给你一个下标从 0 开始、大小为 n x n 的整数矩阵 grid ,返回满足 Ri 行和 Cj 列相等的行列对 (Ri, Cj) 的数目。
如果行和列以相同的顺序包含相同的元素(即相等的数组),则认为二者是相等的。
示例 1:
输入:grid = [[3,2,1],[1,7,6],[2,7,7]]
输出:1
解释:存在一对相等行列对:
- (第 2 行,第 1 列):[2,7,7]
示例 2:
输入:grid = [[3,1,2,2],[1,4,4,5],[2,4,2,2],[2,4,2,2]]
输出:3
解释:存在三对相等行列对:
- (第 0 行,第 0 列):[3,1,2,2]
- (第 2 行, 第 2 列):[2,4,2,2]
- (第 3 行, 第 2 列):[2,4,2,2]提示:
n == grid.length == grid[i].length
1 <= n <= 200
1 <= grid[i][j] <= 10^5
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/equal-row-and-column-pairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
- 暴力,找到首列和首行相同的数,然后再进行比较
class Solution:def equalPairs(self, grid: List[List[int]]) -> int:n = len(grid)m = defaultdict(list) # v: [idx ]for i in range(n):m[grid[0][i]].append(i)def same(x, y):for i in range(n):if grid[x][i] != grid[i][y]:return Falsereturn Trueans = 0for i in range(n):for j in m[grid[i][0]]:if same(i, j):ans += 1return ans
1976 ms 18.6 MB Python3
- 更优的做法,转置后得到的新矩阵 跟原来的进行比较
class Solution:def equalPairs(self, grid: List[List[int]]) -> int:n = len(grid)g2 = [[0 for i in range(n)] for j in range(n)]for i in range(n):for j in range(n):g2[i][j] = grid[j][i]ans = 0for i in range(n):for j in range(n):if g2[i] == grid[j]:ans += 1return ans
404 ms 18.5 MB Python3
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!