前缀和:通常使用前缀和来解决这样的问题,给定一组数组,给定两个数l,r,让求这个数组里面从[l,r]里的数的和,可能你说遍历也能解决,但是当数据量很大时,就会直接TLE,于是这时候我们的前缀和就派上用场了,使用前缀和我们可以将时间复杂度缩短至O(m+n),接下来是求前缀和的过程:
ans=b[r]-b[l-1]
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e6+9;
int a[N];//定义原数组
int b[N];//定义前缀和数组
int main()
{
int m,n;
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];//输入原数组
for(int i=1;i<=n;i++) b[i]=b[i-1]+a[i];//求前缀和数组
while(m--)//假设有m次操作
{
int x,y;
cin>>x>>y;
cout<<b[y]-b[x-1]<<endl;//直接在前缀和数组里面求结果
}
return 0;
}
差分:差分就是先有一个整数序列,接下来输入m个操作,每个操作包含三个整数l, r, c,表示将序列中[l, r]之间的每个数加上c。
最后输出进行完所有操作后的序列。
加+: b[x]=b[x]+z;
b[y+1]=b[y+1]-z;
减-: b[x]=b[x]-z;
b[y+1]=b[y+1]+z;
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<map>
#include<math.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;
const int N=1e5+9;
int a[N];//定义原数组
int b[N];//定义差分数组
int main()
{
int n,m;
cin>>n>>m;
for(int i=1; i<=n; i++) cin>>a[i];//输入原数组
for(int i=1; i<=n; i++) b[i]=a[i]-a[i-1];//求差分数组
while(m--)//m次操作
{
int x,y,z;
cin>>x>>y>>z;
b[x]=b[x]+z;
b[y+1]=b[y+1]-z;
}
for(int i=1; i<=n; i++)//遍历差分数组,求前缀和再输出
{
b[i]=b[i-1]+b[i];
cout<<b[i]<<" ";
}
return 0;
}