【数据结构和算法】 相等行列对
前言
这是力扣的 2352 题，难度为中等，解题方案有很多种，本文讲解我认为最奇妙的一种。
一、题目描述
给你一个下标从 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].length1 <= n <= 2001 <= grid[i][j] <= 105
二、题解
2.1 三层循环
思路与算法：
我们直接将矩阵 gridgridgrid 的每一行和每一列进行比较，如果相等，那么就是一对相等行列对，答案加一。
2.2 哈希 + 二层循环
思路与算法：
这道题暴力解：遍历每一列，然后遍历每一行，再比对当前行和当前列是否以相同顺序包含相同元素。
遍历每一行的时间复杂度为O(n^2)，再套上一层遍历每一列时间复杂度就为O(n^3)。
我们可以发现，我们其实在遍历每一列的时候都在重复的遍历每一行，那么我们可以使用哈希表来存储每一行的数字序列字符串。
然后在遍历每一个行的时候生成这一行对应的数字序列字符串，哈希表中记录有这个数字序列字符串的个数就是对应的行列对个数。
如果直接把数字进行拼接会造成歧义，可能不同的数字会有相同数字序列字符串。因此每一个数字之后添加一个标识符%进行区分。
图解
以示例2进行图解
三、代码
3.1 三层循环
Java版本：
class Solution {
public int equalPairs(int[][] grid) {
int n = grid.length;
int ans = 0;
for (int i = 0; i < n; ++i) {
for (int j = 0; j < n; ++j) {
int ok = 1;
for (int k = 0; k < n; ++k) {
if (grid[i][k] != grid[k][j]) {
ok = 0;
break;
}
}
ans += ok;
}
}
return ans;
}
}C++版本：
class Solution {
public:
int equalPairs(vector<vector<int>>& grid) {
int n = grid.size();
int ans = 0;
for (int i = 0; i < n; ++i) {
for (int j = 0; j < n; ++j) {
int ok = 1;
for (int k = 0; k < n; ++k) {
if (grid[i][k] != grid[k][j]) {
ok = 0;
break;
}
}
ans += ok;
}
}
return ans;
}
};Python版本：
class Solution:
def equalPairs(self, grid: List[List[int]]) -> int:
g = [list(col) for col in zip(*grid)]
return sum(row == col for row in grid for col in g)3.2 哈希 + 二层循环
Java版本：
class Solution {
public int equalPairs(int[][] grid) {
int n = grid.length; // 矩阵尺寸
Map<String, Integer> rowSeqCount = new HashMap<>(); // 存储行数字序列字符串的哈希表
StringBuilder sb ; // 用于生成数字序列字符串
String rowSeq; // 数字序列字符串
for(int i = 0; i < n; i++){ // 遍历每一行
sb = new StringBuilder(); // 每一行新建一个对象
// 生成行数字序列字符串
for(int j = 0; j < n; j++){
sb.append(grid[i][j]);
sb.append('%');
}
rowSeq = sb.toString();
// 哈希表记录这个数字序列字符串个数
rowSeqCount.put(rowSeq, rowSeqCount.getOrDefault(rowSeq, 0) + 1);
}
int count = 0;
for(int j = 0; j < n; j++){ // 遍历每一列
sb = new StringBuilder(); // 每一列新建一个对象
// 生成列数字序列字符串
for(int i = 0; i < n; i++){
sb.append(grid[i][j]);
sb.append('%');
}
rowSeq = sb.toString();
// 从哈希表中查询是和这个列数字序列字符串相同的行数字序列字符串的个数
count += rowSeqCount.getOrDefault(rowSeq, 0);
}
return count;
}
}C++版本：
class Solution {
public:
int equalPairs(std::vector<std::vector<int>>& grid) {
int n = grid.size(); // 矩阵尺寸
std::unordered_map<std::string, int> row_seq_count; // 存储行数字序列字符串的哈希表
for (int i = 0; i < n; i++) { // 用于生成数字序列字符串
std::string row_seq = ""; // 每一行新建一个字符串
// 生成行数字序列字符串
for (int j = 0; j < n; j++) {
row_seq += std::to_string(grid[i][j]) + "%";
}
// 哈希表记录这个数字序列字符串个数
row_seq_count[row_seq] = row_seq_count[row_seq] + 1;
}
int count = 0;
for (int j = 0; j < n; j++) { // 遍历每一列
std::string row_seq = ""; // 每一列新建一个对象
// 生成列数字序列字符串
for (int i = 0; i < n; i++) {
row_seq += std::to_string(grid[i][j]) + "%";
}
// 从哈希表中查询是和这个列数字序列字符串相同的行数字序列字符串的个数
count += row_seq_count[row_seq];
}
return count;
}
};Python版本：
class Solution:
def equalPairs(self, grid: List[List[int]]) -> int:
n = len(grid) # 矩阵尺寸
row_seq_count = {} # 存储行数字序列字符串的哈希表
for i in range(n): # 用于生成数字序列字符串
row_seq = "" # 每一行新建一个字符串
# 生成行数字序列字符串
for j in range(n):
row_seq += f"{grid[i][j]}%"
# 哈希表记录这个数字序列字符串个数
row_seq_count[row_seq] = row_seq_count.get(row_seq, 0) + 1
count = 0
for j in range(n): # 遍历每一列
row_seq = "" # 每一列新建一个对象
# 生成列数字序列字符串
for i in range(n):
row_seq += f"{grid[i][j]}%"
# 从哈希表中查询是和这个列数字序列字符串相同的行数字序列字符串的个数
count += row_seq_count.get(row_seq, 0)
return count四、复杂度分析
4.1 三层循环
时间复杂度： O(n^3)。其中 n 为矩阵 grid 的行数或列数。
空间复杂度： O(1)。
4.2 哈希 + 二层循环
时间复杂度： O(n^2)。其中 n 为矩阵 grid 的行数或列数。
空间复杂度： O(n^2)。
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2024-01-12，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录