LeetCode 2352.相等行列对

题目描述

给你一个下标从 0 开始、大小为 $nxn$ 的整数矩阵 $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$

---

思路

使用哈希表。
这题的难点在于怎么存储key为数组的哈希表，具体看代码部分。
1.遍历行，行存入哈希表。
2.遍历列，如果有与某一行相同的列，返回值++。

C++ 代码

class Solution {
public:
    int equalPairs(vector<vector<int>>& grid) {
        map<vector<int>,int> map;
        // 行加入哈希表
        for (auto x: grid) map[x]++;

        int ret = 0;
        // 遍历列
        for (int j = 0; j < grid.size(); j++)
        {
            vector<int> arr;
            // 列加入arr中
            for (int i = 0; i < grid.size(); i++)
                arr.emplace_back(grid[i][j]);
            // 在行中找到相同的arr 加ret
            if (map.find(arr) != map.end())
                ret+=map[arr];
        }
        return ret;
    }
};