[poj 3468]A Simple Problem with Integers｛树状数组｝

题目

http://poj.org/problem?id=3468

---

解题思路

对于每条指令“C l r d”，把$b[l]$加上$d$，再把$b[r+1]$减去$d$。
$b$数组的前缀和$\sum_{i=1}^{x}b[i]$就是经过这些指令后$a[x]$增加的值。
那么序列$a$的前缀和$a[1\sim x]$整体增加的值就是：

$$\sum_{i=1}^{x}\sum_{j=1}^{i}b[j]$$
上式可以改写为： $$\sum_{i=1}^{x}\sum_{j=1}^{i}b[j]=\sum_{i=1}^{x}(x-i+1)*b[i]=(x+1)\sum_{i=1}^{x}b[i]-\sum_{i=1}^{x}i*b[i]$$
我们可以用一个树状数组，维护 $i*b[i]$的前缀和 $\sum_{i=1}^{x}i*b[i]$，上式就可以直接查询、计算。

---

代码

#include<cstdio>
#include<algorithm>
#include<string>
#include<iostream>
using namespace std; 
string s; int n,m,aa,l,r,d; long long ans,sum[100010],c[2][100010]; 
long long ask(int k,int x){
    long long ans=0; 
    for (;x;x-=x&-x) ans+=c[k][x]; return ans; 
}
void add(int k,int x,int y){
    for (;x<=n;x+=x&-x) c[k][x]+=y; 
}
int main()
{
    scanf("%d%d",&n,&m); 
    for (int i=1;i<=n;i++){
        scanf("%d",&aa); sum[i]=sum[i-1]+aa; 
    }
    for (int i=1;i<=m;i++)
    {
        cin>>s; scanf("%d%d",&l,&r); 
        if (s[0]=='C') {
            scanf("%d",&d); 
            add(0,l,d); add(0,r+1,-d); add(1,l,l*d); add(1,r+1,-(r+1)*d);
        } else{
            ans=sum[r]+(r+1)*ask(0,r)-ask(1,r); 
            ans-=sum[l-1]+l*ask(0,l-1)-ask(1,l-1); 
            printf("%lld\n",ans); 
        }
    }
}