【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;
    }
}
相关推荐
YouEmbedded19 小时前
解码数据结构内核链表
数据结构·链表·内核链表
Han.miracle19 小时前
数据结构——二叉树学习
数据结构·学习
hello_lain20 小时前
9.1 简单排序(冒泡、插入)(排序(上))
c语言·数据结构·算法·排序算法
炬火初现20 小时前
Hot100——普通数组
算法·leetcode·职场和发展
啊我不会诶20 小时前
【数据结构】字典树
数据结构·算法
信奥卷王20 小时前
[GESP202403 五级] 成绩排序
数据结构·算法
小欣加油21 小时前
leetcode 494 目标和
c++·算法·leetcode·职场和发展·深度优先
Jiezcode1 天前
LeetCode 55.跳跃游戏
c++·算法·leetcode·游戏
wheeldown1 天前
【Leetcode高效算法】用双指针策略打破有效三角形的个数
python·算法·leetcode
蒙奇D索大1 天前
【数据结构】数据结构秘籍:如何衡量“查找”的快慢?ASL是关键!
数据结构·笔记·学习·考研