Leetcode 2352. 相等行列对

题目:

给你一个下标从 0 开始、大小为 n x n 的整数矩阵 grid ,返回满足 Ri 行和 Cj 列相等的行列对 (Ri, Cj) 的数目。

如果行和列以相同的顺序包含相同的元素(即相等的数组),则认为二者是相等的。

难度:中等

示例1:

image

输入:grid = [[3,2,1],[1,7,6],[2,7,7]]
输出:1
解释:存在一对相等行列对:
- (第 2 行,第 1 列):[2,7,7]

示例2:

image

输入: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] <= 105

代码实现:

方法一:暴力

class Solution {
public:
    int equalPairs(vector<vector<int>>& grid) {
        int n = grid.size();
        int res = 0;
        // 判断 i 行 j 列 是否相等
        auto check = [&](int i, int j){
            for(int k = 0; k < n; ++k){
                if(grid[i][k] != grid[k][j]) return false;
            }
            return true;
        };
        for(int i = 0; i < n; ++i){
            // 比较第i行 和 第j列
            for(int j = 0; j < n; ++j){
                res += check(i, j);
            }
        }
        return res;
    }
};

方法二:空间匹配
class Solution {
public:
    int equalPairs(vector<vector<int>>& grid) {
        int n = grid.size();
        int res = 0, tmp[n];
        // 遍历列
        for(int j = 0; j < n; ++j){
            // 临时存储列
            for(int i = 0; i < n; ++i){
                tmp[i] = grid[i][j];
            }
            // 遍历行 进行比较
            for(int i = 0; i < n; ++i){
                res += std::equal(grid[i].begin(), grid[i].end(), tmp);
            }

        }
        return res;
    }
};

posted @   DL1024  阅读(11)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】
点击右上角即可分享
微信分享提示