【Leetcode 每日一题】2711. 对角线上不同值的数量差

问题背景

给你一个下标从 0 0 0 开始、大小为 m × n m \times n m×n 的二维矩阵 g r i d grid grid,请你求解大小同样为 m × n m \times n m×n 的答案矩阵 a n s w e r answer answer。

矩阵 a n s w e r answer answer 中每个单元格 ( r , c ) (r, c) (r,c) 的值可以按下述方式进行计算:

  • 令 t o p L e f t [ r ] [ c ] topLeft[r][c] topLeft[r][c] 为矩阵 g r i d grid grid 中单元格 ( r , c ) (r, c) (r,c) 左上角对角线上 不同值 的数量。
  • 令 b o t t o m R i g h t [ r ] [ c ] bottomRight[r][c] bottomRight[r][c] 为矩阵 g r i d grid grid 中单元格 ( r , c ) (r, c) (r,c) 右下角对角线上 不同值 的数量。

然后 a n s w e r [ r ] [ c ] = ∣ t o p L e f t [ r ] [ c ] − b o t t o m R i g h t [ r ] [ c ] ∣ answer[r][c] = |topLeft[r][c] - bottomRight[r][c]| answer[r][c]=∣topLeft[r][c]−bottomRight[r][c]∣。

返回矩阵 a n s w e r answer answer。
矩阵对角线 是从最顶行或最左列的某个单元格开始,向右下方向走到矩阵末尾的对角线。

如果单元格 ( r 1 , c 1 ) (r_1, c_1) (r1,c1) 和单元格 ( r , c ) (r, c) (r,c) 属于同一条对角线且 r 1 < r r_1 < r r1<r,则单元格 ( r 1 , c 1 ) (r_1, c_1) (r1,c1) 属于单元格 ( r , c ) (r, c) (r,c) 的左上对角线。类似地,可以定义右下对角线。

数据约束

  • m = g r i d . l e n g t h m = grid.length m=grid.length
  • n = g r i d [ i ] . l e n g t h n = grid[i].length n=grid[i].length
  • 1 ≤ m , n , g r i d [ i ] [ j ] ≤ 50 1 \le m, n, grid[i][j] \le 50 1≤m,n,grid[i][j]≤50

解题过程

应用类似前后缀分解的思想,按顺序遍历矩阵,分别用集合统计相应的左上和右下的元素数量即可。

进一步追求效率的话,还可以用位运算来模拟集合操作。

具体实现

java 复制代码
class Solution {
    public int[][] differenceOfDistinctValues(int[][] grid) {
        int m = grid.length;
        int n = grid[0].length;
        int[][] res = new int[m][n];
        Set<Integer> set = new HashSet<>();
        for (int k = 1; k < m + n; k++) {
            int min = Math.max(n - k, 0);
            int max = Math.min(m + n - 1 - k, n - 1);
            set.clear();
            for (int j = min; j <= max; j++) {
                int i = k + j - n;
                res[i][j] = set.size();
                set.add(grid[i][j]);
            }
            set.clear();
            for (int j = max; j >= min; j--) {
                int i = k + j - n;
                res[i][j] = Math.abs(res[i][j] - set.size());
                set.add(grid[i][j]);
            }
        }
        return res;
    }
}
相关推荐
快去睡觉~6 小时前
力扣73:矩阵置零
算法·leetcode·矩阵
岁忧6 小时前
(nice!!!)(LeetCode 每日一题) 679. 24 点游戏 (深度优先搜索)
java·c++·leetcode·游戏·go·深度优先
小欣加油6 小时前
leetcode 3 无重复字符的最长子串
c++·算法·leetcode
月盈缺9 小时前
学习嵌入式的第二十二天——数据结构——双向链表
数据结构·学习·链表
猿究院--王升9 小时前
jvm三色标记
java·jvm·算法
一车小面包9 小时前
逻辑回归 从0到1
算法·机器学习·逻辑回归
科大饭桶10 小时前
C++入门自学Day14-- Stack和Queue的自实现(适配器)
c语言·开发语言·数据结构·c++·容器
tt55555555555511 小时前
字符串与算法题详解:最长回文子串、IP 地址转换、字符串排序、蛇形矩阵与字符串加密
c++·算法·矩阵
元亓亓亓11 小时前
LeetCode热题100--101. 对称二叉树--简单
算法·leetcode·职场和发展
躲在云朵里`11 小时前
深入理解数据结构:从数组、链表到B树家族
数据结构·b树