【算法题】2352. 相等行列对

插: 前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站。
坚持不懈,越努力越幸运,大家一起学习鸭~~~

题目:

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

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

示例 1:
【算法题】2352. 相等行列对_第1张图片
输入:grid = [[3,2,1],[1,7,6],[2,7,7]]
输出:1
解释:存在一对相等行列对:

  • (第 2 行,第 1 列):[2,7,7]

示例 2:
【算法题】2352. 相等行列对_第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

java代码:

class Solution {
    public int equalPairs(int[][] grid) {
        Map map = new HashMap<>();
        for (int[] ints : grid) {
            map.put(Arrays.toString(ints), map.getOrDefault(Arrays.toString(ints), 0) + 1); // 将每行散列一下。
        }
        int[][] grid2 = new int[grid[0].length][grid.length]; // 没必要这么麻烦,行列长度相等。
        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[0].length; j++) {
                grid2[j][i] = grid[i][j]; // 行列转置
            }
        }
        int res = 0;
        for (int[] ints : grid2) {
            if (map.containsKey(Arrays.toString(ints))) {
                res += map.get(Arrays.toString(ints)); // 与当前列所相等的行数。
            }
        }
        return res;
    }
}

你可能感兴趣的:(算法,数据结构,c++)