前缀和算法

(一)前缀和算法

概念:前缀和就是数组的前i项之和

一维前缀和

s[1]=a[1]

s[2]=a[1]+a[2]

s[3]=a[1]+a[2]+a[3]

s[4]=a[1]+a[2]+a[3]+a[4]

s[5]=a[1]+a[2]+a[3]+a[4]+a[5]

①.前缀和

输入一个长度为n的整数序列。

接下来再输入m个询问,每个询问输入一对l, r。

对于每个询问,输出原序列中从第l个数到第r个数的和。

输入格式

第一行包含两个整数n和m。

第二行包含n个整数,表示整数数列。

接下来m行,每行包含两个整数l和r,表示一个询问的区间范围。

输出格式

共m行,每行输出一个询问的结果。

数据范围

1lrn,
1n,m100000,
10001000

输入样例:

5 3
2 1 3 6 4
1 2
1 3
2 4

输出样例:

3
6
10
#include<iostream>
using namespace std;
int main()
{
    int sum,n,i,j,x,m,l,r;
    int s[100009];
    cin>>n>>m;
    sum=0;
    for(i=1;i<=n;i++)
    {
        cin>>x;
        sum+=x;
        s[i]=sum;
    }
    for(i=0;i<m;i++)
    {
        cin>>l>>r;
        cout<<s[r]-s[l-1]<<endl;
    }
    return 0;
}
 

在做题中我们会经常用到他的二维的前缀和数组

最常见的案例:求子矩阵的和

2.子矩阵的和

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

对于每个询问输出子矩阵中所有数的和。

输入格式

第一行包含三个整数n,m,q。

接下来n行,每行包含m个整数,表示整数矩阵。

接下来q行,每行包含四个整数x1, y1, x2, y2,表示一组询问。

输出格式

共q行,每行输出一个询问的结果。

数据范围

1n,m1000,
1q200000,
1x1x2n,
1y1y2m,
10001000

输入样例:

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

对于二维前缀和问题,首先我们先求出他的前缀和数组,如何求呢?
对于(i,j)该点的前缀和数组求解:看图,求红色部分的前缀和

 就可以看成是蓝色前缀和+粉色前缀和-黄色前缀和+自己本身

s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j]

 求完前缀和数组后,在分析一下如果要求两个点之间的子矩阵的和应该怎么办?

再画图分析一下:

 求解两个圆圈之间的子矩阵:黄色面积-蓝色面积-紫色面积+粉色面积

ans=s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]

代码:

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,q,x1,x2,y1,y2;
int s[1010][1010],a[1010][1010];
int main()
{
    int i,j;
    scanf("%d%d%d",&n,&m,&q);
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
        }
    }
    while(q--)
    {
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        printf("%d
",s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]);
    }
    return 0;
}
 
原文地址:https://www.cnblogs.com/xiaofengzai/p/12045909.html