【算法】前缀和

最新推荐文章于 2024-10-17 11:39:06 发布

lll耶耶耶

最新推荐文章于 2024-10-17 11:39:06 发布

阅读量238

点赞数 5

文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_62771564/article/details/142517292

版权

概念及原理

简而言之就是快速求出数组中某一段连续区间的和

具体原理可参考该篇博客：

前缀和与差分图文并茂超详细整理（全网最通俗易懂）-CSDN博客

模板

一维前缀和

1.预处理一个前缀和数组sum

sum[i] :表示[1，i]区间内所有元素的和

sum[i] = sum[i - 1] +arr[i](画图理解)

2.根据题目要求使用前缀和数组

例如：返回[l,r]区间内的元素和 --->sum[r] - sum[l - 1]

PS:下标要从1开始计数，是为了处理边界情况

经典例题：【模板】前缀和_牛客题霸_牛客网 (nowcoder.com)

public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        
        //1、输入两行整数和数组
        int n = in.nextInt();
        int q = in.nextInt();
        int arr[] = new int[n+1];
        for(int i = 1; i <= n;i++){
            arr[i] = in.nextInt();
        }
        arr[0] = 0;
        
        //2、预处理一个前缀和数组
        //这里处理了边界情况，使下标从1开始
        long[] dp = new long[n+1];
        dp[0] = 0;
        for(int i = 1;i <= n;i++){
            dp[i] = dp[i-1]+arr[i];
        }

        //3、使用前缀和数组
        while(q > 0){
            int l = in.nextInt();
            int r = in.nextInt();
            System.out.println(dp[r] - dp[l-1]);
            q--;
        }


    }

二维前缀和

1.预处理一个前缀和数组矩阵sum

sum[i][j] :表示[1，1]位置到[i,j]位置，这段区间内所有元素的和


sum[i][j] = sum[i-1][j]+sum[i][j-1]+arr[i][j]-sum[i-1][j-1]


2.根据题目要求使用前缀和数组[画图]

例如：返回[x1,y1] ~ [x2,y2]区间内的元素和

 ---> sum[x2][y2] - sum[x1- 1][y2] - sum[x2][y1-1]+sum[x1-1][y1-1]


PS:下标要从1开始计数，是为了处理边界情况

对应第一点的图：

对应第二点的图：

【两张图都是上面分享的博客中的截取的照片】

经典例题：【模板】二维前缀和_牛客题霸_牛客网 (nowcoder.com)

public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
       //读入数据
       int n = in.nextInt();
       int m = in.nextInt();
       int q = in.nextInt();
       int[][] arr = new int[n+1][m+1];
       for(int i = 1;i <= n;i++){
            for(int j = 1; j <= m;j++){
                arr[i][j] = in.nextInt();
            }
       }

       //2、预处理一个前缀和矩阵
       long[][] dp =new long[n+1][m+1];
       for(int i = 1;i <= n;i++){
        for(int j = 1;j <= m;j++){
            dp[i][j] = dp[i-1][j]+dp[i][j-1] +arr[i][j] -dp[i-1][j-1];
        }
       }
       //3、使用前缀和矩阵
       while(q > 0 ){
            int x1 = in.nextInt();
            int y1 = in.nextInt();
            int x2 = in.nextInt();
            int y2 = in.nextInt();
            System.out.println(dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1]);
            q--;
       }
    }