lc2536.子矩阵元素加1

**暴力解法:**直接按照题目所示在矩阵的相应位置加一

时间复杂度:O(n 2 * queries.length)

空间复杂度:O(1)

**二维差分:**创建二维差分数组,通过对差分数组的修改来影响原来的数组,最后还原

时间复杂度:O(n 2 + queries.length)

空间复杂度:O(n 2 )

示例2此种情况会发生角标越界的情况,因此差分数组需要多初始化2行2列

代码

复制代码
import org.junit.Test;

public class SubmatrixPlus {

    @Test
    public void test() {
        int[][] queries = new int[][]{{1, 1, 2, 2}, {0, 0, 1, 1}};
        for (int[] query : submatrixPlus(queries, 3)) {
            for (int n : query) {
                System.out.print(n + " ");
            }
            System.out.println();
        }

        int[][] queries1 = new int[][]{{0, 0, 1, 1}};
        for (int[] query : submatrixPlus(queries1, 2)) {
            for (int n : query) {
                System.out.print(n + " ");
            }
            System.out.println();
        }
    }

    //int[][] querries = {{左上角行,左上角列,右下角行,右下角列},{左上角行,左上角列,右下角行,右下角列}}
    public static int[][] submatrixPlus(int[][] queries, int n) {
        // 构建差分数组,多初始化2行2列避免数组越界
        int[][] arr = new int[n][n];
        for (int i = 0; i < queries.length; i++) {
            arr[queries[i][0] + 1][queries[i][1] + 1]++;//第几行不等于数组的索引
            arr[queries[i][2] + 2][queries[i][1] + 1]--;
            arr[queries[i][0] + 1][queries[i][3] + 2]--;
            arr[queries[i][2] + 2][queries[i][3] + 2]++;
        }
        //还原数组
        int[][] res = new int[n + 2][n + 2];
        for (int i = 0; i < res.length; i++) {
            for (int j = 0; j < res[i].length; j++) {
                arr[i + 1][j + 1] = arr[i + 1][j + 1] + arr[i + 1][j] + arr[i][j + 1] - arr[i][j];
                res[i][j] = arr[i + 1][j + 1];
            }
        }
        return res;

    }
}
相关推荐
牛客企业服务9 分钟前
2025年AI面试推荐榜单,数字化招聘转型优选
人工智能·python·算法·面试·职场和发展·金融·求职招聘
糖葫芦君1 小时前
Policy Gradient【强化学习的数学原理】
算法
向阳@向远方3 小时前
第二章 简单程序设计
开发语言·c++·算法
github_czy3 小时前
RRF (Reciprocal Rank Fusion) 排序算法详解
算法·排序算法
许愿与你永世安宁4 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子4 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
满分观察网友z4 小时前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法
YuTaoShao5 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵
Heartoxx5 小时前
c语言-指针(数组)练习2
c语言·数据结构·算法
大熊背5 小时前
图像处理专业书籍以及网络资源总结
人工智能·算法·microsoft