其实这题应该是线段树区间更新的模板题,我现在用树状数组的方法做
理解树状数组一维的区间更新和区间查询需要一些基础(知道的直接继续往下看):
树状数组 一维 单点更新 区间查询(最基础):http://blog.csdn.net/williamsun0122/article/details/70679766
树状数组 一维 区间更新 单点查询:http://blog.csdn.net/williamsun0122/article/details/71511879
其实树状数组本质就是单点更新、区间查询,其它用法都是通过转化成本质用法实现
一维的区间更新和单点查询是通过维护一个差分数组转化实现,而一维的区间更新和区间查询是通过维护一个差分数组和一个与差分数组有关的数组转化实现。
设原数组是a[n],差分数组c[n],c[i]=a[i]-a[i-1],那么明显地a[n]= ∑ni=1c[i] ,如果想要修改a[i]到a[j] (比如+v),只需令c[i]+=v,c[j+1]-=v即可,到这区间更新就实现了。
那么怎么实现区间查询呢?(比如查询a[1]-a[n])
a[1] + a[2] + … + a[n]
= c[1] + (c[1]+c[2]) + … + (c[1]+c[2]+…+c[n])
= n*c[1] + (n-1)*c[2] + … + c[n]
= (n+1)*(c[1]+…+c[n]) - (1*c[1] + 2*c[2] + … + n*c[n])
所以我们在维护差分数组c[]的同时维护与之相关的数组c2[n] = n*c[n]即可
以poj3468为例,再看看代码注释应该就懂了。
//#include <bits/stdc++.h>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
ll a[maxn],c1[maxn],c2[maxn],n,q;
ll lowbit(ll x)
{
return x&(-x);
}
void update(ll x,ll value)
{
ll i=x;
while(x<=n)
{
c1[x] += value;
c2[x] += i*value;
//更新c1[]的同时更新c2[]
x += lowbit(x);
}
}
ll query(ll x)
{
ll tmp1=0,tmp2=0,i=x;
while(x>0)
{
tmp1 += c1[x];
tmp2 += c2[x];
x -= lowbit(x);
}
return (i+1)*tmp1-tmp2;
//tmp1 = c1[1]+...+c1[n],tmp2 = c2[1]+...+c2[n] = 1*c1[1]+...+n*c1[n]
//返回a[1]+...+a[x]
}
int main()
{
ll tmp;
scanf("%d%d",&n,&q);
memset(a,0,sizeof(a));
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
for(int i=1;i<=n;i++)
{
scanf("%lld",a+i);
//初始c1[]和c2[],单点更新
update(i,a[i]-a[i-1]);
}
int a,b;
char op;
while(q--)
{
getchar();
scanf("%c%d%d",&op,&a,&b);
if(op=='C')
{
scanf("%lld",&tmp);
//区间更新a[],通过两次单点更新c1[]实现,更新c1[]的同时更新c2[]
update(a,tmp);
update(b+1,-tmp);
}
else
{
printf("%lld\n",query(b)-query(a-1));
//输出a[a]+...+a[b]
}
}
return 0;
}