前缀和算法（一维、二维）

算法思想

能够快速的求某一段的和，查询的时候效率很高
只能查询，不能修改

一维前缀和

思路

例题：前缀和

题目信息

题解

#include <bits/stdc++.h>
#define int long long
#define maxsize 100000
using namespace std;

int n,m;
int a[maxsize];
int s[maxsize];

signed main()
{
    int l,r;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    for(int i=1;i<=n;i++)
    {
        s[i]=s[i-1]+a[i];
    }
    for(int i=0;i<m;i++)
    {
        cin>>l>>r;
        cout<<s[r]-s[l-1]<<endl;
    }
    return 0;
}

二维前缀和

思路

例题：子矩阵的和

题目信息

题解

#include <bits/stdc++.h>
#define int long long
#define maxsize 1000
using namespace std;

int n,m,q;
int a[maxsize][maxsize];
int s[maxsize][maxsize];

signed main()
{
    int x1,x2,y1,y2;
    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]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
        }
    }
    for(int i=0;i<q;i++)
    {
        cin>>x1>>y1>>x2>>y2;
        cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<<endl;
    }
    return 0;
}