第一篇博客,聊聊第一个学会的基础算法---前缀和
先来看一道题
例题引入
input
第一行输入n,m。
接下来一行n个数字,第i个数字为ai。
接下来m行,每行一个询问,输入x,y,z。
output
对于每一行询问,
若x=1,输出ay到az的和。
看到这题的时候不知道大家的想法的是不是直接定义俩个数组,一个存放数据,另一个当输入的x=1时,让存放数据的数组的第y项到z项依次加给这个数组,最后就得出了结果。
可这样的方法过于复杂,会在运行时超时。
像数学题一样当我们遇到这种复杂的题型时总是有简便方法,那么我们来思考一样,这道题的简便方法是什么?就是我们的前缀和,想象一下我们要算第y到z项的和,可不可以先将前z项的和算出来,再把前y项的和算出来,最后用前z项减去前y项就得到了y第y项到第z项的和。
首先我们得把每一项的之前的和算出来:
我们会发现新数组s【】中的每一项都等于a【】数组中的前几项:
所以根据上面的流程我们发现这样一个公式;
接下来我们再来看一到例题:
description
有一条数轴,初始所有值都为零。 n次操作给每个点加上一个值,求所有w宽度的区间的区间和最大的区间
input
一行 N,W,分别代表操作的数量和w
余下N 行,输入 Xi 和Bi,代表给Xi加上Bi。
output
一个数字,代表宽度为w的区间和最大的区间
对于 10%10% 的数据,W=0(没有边缘);
对于 40%40% 的数据,W≤1000;
对于 100%100% 的数据,1≤N≤105,0≤W≤105,1≤Xi≤105,1≤Bi≤100。
除 W=0 的情况外,W 均为 ≥3≥3 的奇数。
#include<bits/stdc++.h>
using namespace std;
const int N = 100005;
int c[N];//树状数组,这道题只是区间求和,不用存星星,所以不用再开一个星星数组
int n;
int w;
int Max = -1;//等下区间求和的时候比较最大值用
int x, y, z;
int lowbit(int x)//lowbit操作//lowbit(x)是x的二进制表达式中最低位的1所对应的值
{
return x & (-x);
}
void update(int x, int v)//update操作//有相同的就覆盖
{
while (x <= n)
{
c[x] += v;
x += lowbit(x);
}
}
int sum(int x)//区间求和操作
{
int res = 0;
while (x > 0)
{
res += c[x];
x -= lowbit(x);
}
return res;
}
int main()
{
scanf("%d%d", &n, &w);
for (int i = 1; i <= n; i++)//读入
scanf("%d%d", &x, &y), update(x, y);
for (int i = w; i <= n; i++)//因为这里不想写n,于是我写了100000
{
Max = max(Max, sum(i - 1) - sum(i - w - 1));//区间最大值
}
printf("%d\n", Max);//输出
return 0;
}