前缀和(一维+二维) 板子

前缀和的数组用全局变量存

一维前缀和

模板题

#include<iostream>
using namespace std;

const int N=100010;

int a[N],s[N];

int main(){
    int n,m;
    while(cin>>n>>m){
        for(int i=1;i<=n;i++){// 从1 开始
            cin>>a[i];
        }
        for(int i=1;i<=n;i++){// 从1 开始
            s[i]=s[i-1]+a[i];
        }
        while(m--){
            int x,y;
            cin>>x>>y;
            cout<<s[y]-s[x-1]<<endl;
        }
    }
}

二维前缀和 即求子矩阵

#include<iostream>
using namespace std;

const int N=1010;
int a[N][N];
int s[N][N];

int main(){
	int m,n,q;
	while(cin>>n>>m>>q){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				cin>>a[i][j];
			}
		}
		// 求前缀和矩阵 
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				s[i][j]=a[i][j]+s[i][j-1]+s[i-1][j]-s[i-1][j-1];
			}
		}
		// 计算子矩阵的和 
		while(q--){
			int x1,y1,x2,y2;
			cin>>x1>>y1>>x2>>y2;
			cout<<s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]<<endl;
		}
	}
}