前缀和模板

一维模板：

例题1：

输入一个长度为 n 的整数序列。
接下来再输入 m 个询问，每个询问输入一对 l,r。
对于每个询问，输出原序列中从第 l 个数到第 r 个数的和。

输入格式
第一行包含两个整数 n 和 m。
第二行包含 n 个整数，表示整数数列。
接下来 m 行，每行包含两个整数 l 和 r，表示一个询问的区间范围。

输出格式
共 m 行，每行输出一个询问的结果。

数据范围
1≤l≤r≤n,
1≤n,m≤100000,
−1000≤数列中元素的值≤1000
输入样例：
5 3
2 1 3 6 4
1 2
1 3
2 4
输出样例：
3
6
10

思想：

1.构建一个前缀和数组来存储啊a₁ ~ a_n的前缀和s_n，s[j]=s[j-1]+a_j;
2.当需要计算i~ j这段数组的和时，可以用s[j]-s[i-1]代替遍历i ~ j直接算出 i~j的的和，时间复杂度由O(n)变为O(1)；

模板：

import java.io.*;
import java.util.*;
public class Main{
    static int[] s;
    public static void main(String[] args){
        Scanner sc =new Scanner(System.in);
        int n=sc.nextInt();
        int m=sc.nextInt();
        s=new int[n+1];
        for(int i=1;i<=n;i++)
            s[i]=s[i-1]+sc.nextInt();//计算前缀和
        while(m-->0){
            int i,j;
            i=sc.nextInt();
            j=sc.nextInt();
            System.out.println(s[j]-s[i-1]);//计算i~j区间的he
        }
          
    }
}

二维模板：

例题2：

输入一个 n 行 m 列的整数矩阵，再输入 q 个询问，每个询问包含四个整数 x1,y1,x2,y2，表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。

输入格式
第一行包含三个整数 n，m，q。
接下来 n 行，每行包含 m 个整数，表示整数矩阵。
接下来 q 行，每行包含四个整数 x1,y1,x2,y2，表示一组询问。

输出格式
共 q 行，每行输出一个询问的结果。

数据范围
1≤n,m≤1000,
1≤q≤200000,
1≤x1≤x2≤n,
1≤y1≤y2≤m,
−1000≤矩阵内元素的值≤1000
输入样例：
3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4
输出样例：
17
27
21

思想：

1.创建一个二维数组来粗存储a_1,1 ~ a_i,j的前缀和，s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a_i,j;

2.当需要计算a_x1,y1 ~ a_x2,y2 区间的和时，使用sum=s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1] 即可代替遍历，将时间复杂度由O(n²) 降为 O(1)。

模板：

import java.util.*;
public class Main{
    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[][] s=new int[n+1][m+1];
        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]+sc.nextInt();//构建前缀和
        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[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]);//计算区间和
        }
        
    }
}