算法前缀和

 

题目链接：

https://ac.nowcoder.com/acm/problem/226282
来源：牛客网

输入描述

给定一个长度为n的数组a1,a2,....an，

接下来有q次查询, 每次查询有两个参数l, r.

对于每个询问, 请输出al+al+1+....+ar

 

输出描述：

输出q行,每行代表一次查询的结果.

示例：

输入：

3 2

1 2 4

1 2

2 3

输出：

3 6

 

方法一：暴力法

这是我们特别容易想到的方法，但是他不是一个好方法，因为暴力法的时间复杂度往往都会很高，在我们算法体重往往会因为超市而不予通过，暴力法可以说是一种无奈之举，即没有更优化的方式解决某个问题，但是问题又不能不解决，这时候我们才能使用暴力法

思路：

0、输入一组数的个数和查询次数

1、输入一组数据存入到数组中

2、循环查询次数，每次都输入l和r,遍历l和r及其之间的数据，计算之和，将和输出

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int numCount = sc.nextInt();
        int queryCount = sc.nextInt();
        // 用来存储一组数据
        int[] arr = new int[numCount];
        // 存储数据到数组中
        for (int i = 0; i < numCount; i++) {
            arr[i] = sc.nextInt();
        }
        // 第一层循环代表查询次数
        for (int i = 0; i < queryCount; i++){
            long sum = 0;
            int r = sc.nextInt();
            int l = sc.nextInt();
            // 第二层表示每次查询的时候计算从l到r的数据之和是多少
            for (int j = r; j <= l; j++) {
                sum += arr[j - 1];
            }
            System.out.println(sum);
        }
    }
}

时间复杂度o(n*m)，很容易超时。

 

方法二：前缀和

    时间复杂度o(n+m)，提高运算效率。

     一维前缀和

思路：每次输入一个数就求一次和，即定义一个sum[]，进行预处理，sum[i]表示数组中前i个数的和，sum[i]=sum[i-1]+arr[i]。对于每次查询，只需计算

sum[R]-sum[L-1] (时间复杂度o(1)) 即可。比如，对于数组{1，2，3，4，5}，要查询第2个数到第4个数的和，

其中sum[0] = 0；

sum[1]=1；

sum[2]=1+2；

sum[3] = 1+2+3;

sum[4]=1+2+3+4;

那么第2个数到第4个数的和就是 2+3+4 ,也就是sum(4) - sum(2-1)

优势：提升时间复杂度，不需要进行两层循环，而暴力法是需要两层循环的

代码如下：

import java.util.Scanner;
public class Main {
    static long[] sum = new long[100001];//初始值为0
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int q = sc.nextInt();
        int[] arr = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            arr[i] = sc.nextInt();
            sum[i] = sum[i - 1] + arr[i];
        }
        for (int i = 0; i < q; i++) {
            int l = sc.nextInt();
            int r = sc.nextInt();
            System.out.println(sum[r]-sum[l - 1]);

        }
    }
}

二维前缀和

链接：https://ac.nowcoder.com/acm/problem/226333
来源：牛客网

题目描述：
给你一个 n 行 m 列的矩阵 A ，下标从1开始。
接下来有 q 次查询，每次查询输入 4 个参数 x1 , y1 , x2 , y2
请输出以 (x1, y1) 为左上角 , (x2,y2) 为右下角的子矩阵的和，

输入描述

输出描述：

输出q行,每行表述查询结果.

示例
3 4 3

1 2 3 4

3 2 1 0

1 5 7 8

1 1 2 2

1 1 3 3

1 2 3 4

输出：

8

25

32

【模板】二维前缀和 (nowcoder.com)

思路：

每输入一个数就求一次和，即定义一个二维数组sum[][]，

sum[i][j]表示坐标为(i,j)的数到坐标为(1,1)的数所覆盖范围中所有数的和，

此时计算(x1,y1)到 (x2,y2)的和只需要满足如下公式：

sum[i][j] - sum[i-1][j] - sum[i][j-1] +sum[i-1][j-1] ;

代码如下：

import java.util.Scanner;

public class Main {
    static long[][] sum = new long[1005][1005];

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int q = sc.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] = sc.nextInt();
                sum[i][j] = sum[i - 1][j]+ sum[i][j - 1] - sum[i - 1][j - 1]+ arr[i][j];

            }
        }
        for (int i = 0; i < q; i++) {
            int x1 = sc.nextInt();
            int y1 = sc.nextInt();
            int x2 = sc.nextInt();
            int y2 = sc.nextInt();
            System.out.println(sum[x2][y2]- sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1]);

        }
    }
}