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;

    }
}
相关推荐
飞川撸码8 分钟前
【LeetCode 热题100】240:搜索二维矩阵 II(详细解析)(Go语言版)
leetcode·矩阵·golang
半盏茶香9 分钟前
启幕数据结构算法雅航新章,穿梭C++梦幻领域的探索之旅——堆的应用之堆排、Top-K问题
java·开发语言·数据结构·c++·python·算法·链表
小竹子1431 分钟前
L1-1 天梯赛座位分配
数据结构·c++·算法
董董灿是个攻城狮41 分钟前
Transformer 通关秘籍8:词向量如何表示近义词?
算法
独好紫罗兰1 小时前
洛谷题单2-P5712 【深基3.例4】Apples-python-流程图重构
开发语言·python·算法
uhakadotcom1 小时前
NVIDIA Resiliency Extension(NVRx)简介:提高PyTorch训练的容错性
算法·面试·github
梭七y2 小时前
【力扣hot100题】(020)搜索二维矩阵Ⅱ
算法·leetcode·职场和发展
v维焓2 小时前
C++(思维导图更新)
开发语言·c++·算法
ylfhpy2 小时前
Java面试黄金宝典22
java·开发语言·算法·面试·职场和发展
Phoebe鑫2 小时前
数据结构每日一题day9(顺序表)★★★★★
数据结构·算法