【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;
    }
}
相关推荐
im_AMBER15 分钟前
Leetcode 38
笔记·学习·算法·leetcode
Miraitowa_cheems1 小时前
LeetCode算法日记 - Day 82: 环形子数组的最大和
java·数据结构·算法·leetcode·决策树·线性回归·深度优先
Code_Shark1 小时前
AtCoder Beginner Contest 426 题解
数据结构·c++·算法·数学建模·青少年编程
仰泳的熊猫1 小时前
LeetCode:698. 划分为k个相等的子集
数据结构·c++·算法·leetcode
豐儀麟阁贵1 小时前
4.5数组排序算法
java·开发语言·数据结构·算法·排序算法
Shinom1ya_2 小时前
算法 day 32
算法
WBluuue3 小时前
数据结构与算法:摩尔投票算法
c++·算法·leetcode
2401_858286113 小时前
OS36.【Linux】简单理解EXT2文件系统(2)
linux·运维·服务器·数据结构·文件系统·ext2
小羊学伽瓦3 小时前
【Java数据结构】——常见力扣题综合
java·数据结构·leetcode·1024程序员节
文火冰糖的硅基工坊4 小时前
[人工智能-大模型-66]:模型层技术 - 两种编程范式:数学函数式编程与逻辑推理式编程,构建起截然不同的智能系统。
人工智能·神经网络·算法·1024程序员节