一维 二维求前缀和、差分

简介: 一维 二维求前缀和、差分

一维前缀和


活动 - AcWing


S[i] = a[1] + a[2] + ... a[i]

a[l] + ... + a[r] = S[r] - S[l - 1]

 final static int N=100010;
    public static void main(String[] args) {
        int []arr=new int[N];
        int []sum=new int[N];
        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt();
        int m=sc.nextInt();
        for (int i = 1; i <=n; i++) {
            arr[i]= sc.nextInt();
        }
        for (int i = 1; i <=n; i++) {
            sum[i]=sum[i-1]+arr[i];
        }
        while (m--!=0){
            int l=sc.nextInt();
            int r=sc.nextInt();
            System.out.println(sum[r]-sum[l-1]);
        }
    }


二维前缀和(子矩阵求和)


活动 - AcWing


S[i, j] = 第i行j列格子左上部分所有元素的和

以(x1, y1)为左上角,(x2, y2)为右下角的子矩阵的和为:

S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]


final static int N=1010,M=1010;
    public static void main(String[] args) {
    int [][]arr=new int[N][M];
    int [][]s=new int[N][M];
    Scanner sc=new Scanner(System.in);
    int n=sc.nextInt();
    int m=sc.nextInt();
    int q=sc.nextInt();
        for (int i = 1; i <=n; i++) {
            for (int j =1; j <=m; j++) {
                arr[i][j]= sc.nextInt();
            }
        }
        for (int i = 1; i <=n; i++) {
            for (int j = 1; j <=m; j++) {
                s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+arr[i][j];
            }
        }
        while (q--!=0){
            int x1=sc.nextInt();
            int y1=sc.nextInt();
            int x2=sc.nextInt();
            int y2=sc.nextInt();
            System.out.println(s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]);
        }
    }

一维差分


活动 - AcWing

给区间[l, r]中的每个数加上c:B[l] += c, B[r + 1] -= c


    final static int N=100010;
    static int [] s=new int[N];
    public static void main(String[] args) {
        int []arr=new int[N];
      Scanner sc=new Scanner(System.in);
      int n=sc.nextInt(),m= sc.nextInt();
        for (int i = 1; i <=n; i++) {
            arr[i]= sc.nextInt();
        }
        for (int i = 1; i <=n; i++) {
            insert(i,i,arr[i]);
        }
        while (m--!=0){
            int l=sc.nextInt(),r=sc.nextInt(),c=sc.nextInt();
            insert(l,r,c);
        }
        for (int i=1; i <=n; i++)  s[i]+=s[i-1];
        for(int i=1;i<=n;i++)  System.out.print(s[i]+" ");
        System.out.println();
    }
    public static void insert(int l,int r,int c){
        s[l]+=c;
        s[r+1]-=c;
    }


二维差分


活动 - AcWing


给以(x1, y1)为左上角,(x2, y2)为右下角的子矩阵中的所有元素加上c:

S[x1, y1] += c, S[x2 + 1, y1] -= c, S[x1, y2 + 1] -= c, S[x2 + 1, y2 + 1] += c


    final static int N=1010;
    static int [][]S=new int[N][N];
    public static void main(String[] args) {
    Scanner sc=new Scanner(System.in);
    int [][]arr=new int[N][N];
    int n= sc.nextInt(),m= sc.nextInt(),q= sc.nextInt();
        for (int i =1; i <=n; i++) {
            for (int j =1; j <=m; j++) {
                arr[i][j]= sc.nextInt();//读入原始数组
                insert(i,j,i,j,arr[i][j]);//构建差分数组S
            }
        }
        while (q-->0){
            int x1= sc.nextInt(),y1= sc.nextInt(),x2= sc.nextInt(),y2= sc.nextInt(),c= sc.nextInt();
            insert(x1,y1,x2,y2,c);
        }
        for (int i =1; i <=n; i++) {
            for (int j =1; j <=m; j++) {
               S[i][j]+=S[i][j-1]+S[i-1][j]-S[i-1][j-1];//求差分数组的前n项和
                System.out.print(S[i][j]+" ");//输出最后结果
            }
            System.out.println();
        }
    }
    public static void insert(int x1,int y1,int x2,int y2,int c){
        S[x1][y1]+=c;
        S[x2+1][y1]-=c;
        S[x1][y2+1]-=c;
        S[x2+1][y2+1]+=c;
    }
相关文章
|
16天前
|
存储 人工智能 BI
差分与前缀和
差分与前缀和
12 0
|
22天前
|
机器学习/深度学习 存储 人工智能
利用前缀和计算二维矩阵子矩阵的和
利用前缀和计算二维矩阵子矩阵的和
17 0
|
8月前
|
存储 NoSQL
第3章 数组与矩阵——3.5 稀疏矩阵
第3章 数组与矩阵——3.5 稀疏矩阵
|
8月前
|
机器学习/深度学习 存储 人工智能
第3章 数组与矩阵——3.2 矩阵操作
第3章 数组与矩阵——3.2 矩阵操作
|
8月前
|
机器学习/深度学习 前端开发 rax
第3章 数组与矩阵——3.4 矩阵运算(1)
第3章 数组与矩阵——3.4 矩阵运算(1)
|
9月前
|
存储 人工智能 算法
【高精度加减乘除法、一维二维前缀和&&差分】思路讲解及代码实现
用一篇Blog来讲解下最近学到的高精度加减乘除法、一维二维前缀和&&差分等算法,为日后的刷题打下坚实的基础。
54 0
|
11月前
前缀和:求子矩阵之和
前缀和:求子矩阵之和
32 0
|
人工智能 算法 安全
【基础算法】差分的应用(一维差分和二维差分)
【基础算法】差分的应用(一维差分和二维差分)
155 0
【基础算法】差分的应用(一维差分和二维差分)
|
人工智能 Java 算法框架/工具
二维前缀和数组&二维差分数组
二维差分数组div中的每一个格子记录的是「以当前位置为区域的左上角(区域右下角恒定为原数组的右下角)的值的变化量」【应该不固定 可以倒转】
290 0
二维前缀和数组&二维差分数组
|
人工智能 JavaScript 算法
前缀和 差分 C++ 小结
前缀和 差分 C++ 小结
83 0
前缀和 差分 C++ 小结