【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;
    }
}
相关推荐
秋难降16 分钟前
栈:从基础概念到实战解题(详细)
数据结构·算法·排序算法
归云鹤18 分钟前
QT信号和槽怎么传输自己定义的数据结构
开发语言·数据结构·qt
ezl1fe31 分钟前
RAG 每日一技(十):向量检索的“死穴”?用混合搜索(Hybrid Search)来拯救!
后端·算法
Ahu_iii34 分钟前
【图论基础】理解图的“闭环”:Tarjan 强连通分量算法全解析
算法·图论
PixelMind36 分钟前
【IQA技术专题】DISTS代码讲解
图像处理·人工智能·python·算法·iqa
项目申报小狂人1 小时前
2025年1中科院1区顶刊SCI-投影迭代优化算法Projection Iterative Methods-附完整Matlab免费代码
开发语言·算法·matlab
AI 嗯啦1 小时前
机械学习--逻辑回归
算法·机器学习·逻辑回归
山烛1 小时前
逻辑回归详解:从数学原理到实际应用
python·算法·机器学习·逻辑回归
爱煲汤的夏二2 小时前
扩展卡尔曼滤波器 (EKF) 与无人机三维姿态估计:从理论到实践
单片机·嵌入式硬件·算法·无人机
sali-tec2 小时前
C# 基于halcon的视觉工作流-章27-带色中线
开发语言·人工智能·算法·计算机视觉·c#