P4868 Preprefix sum

题目描述

输入格式

第一行给出两个整数N，M。分别表示序列长度和操作个数
接下来一行有N个数，即给定的序列a1,a2,…an
接下来M行，每行对应一个操作，格式见题目描述

输出格式

对于每个询问操作，输出一行，表示所询问的SSi的值。

样例

输入
5 3
1 2 3 4 5
Query 5
Modify 3 2
Query 5
输出
35
32

说明/提示

1<=N,M<=100000,且在任意时刻0<=Ai<=100000

题目分析

这也是一道比较典型的用树状数组来做的题目。
这个题的操作有两个：单点修改、查询前缀和的前缀和SS[i]。
修改操作可以直接用树状数组来完成，但是查询操作光靠一个树状数组是不够的。
设：S[i]为a[i]的前缀和,SS[i]为S[i]的前缀和。
S[1]：a[1]
S[2]：a[1] a[2]
S[3]：a[1] a[2] a[3]
……
S[x]：a[1] a[2] a[3] a[4]……a[x]
上面所有的a[]全部加起来才能得到SS[x]的值。由上面的方阵我们可以得到：
  a[1] a[2] a[3] a[4]……a[x]
S[1]：a[1] a[2] a[3] a[4]……a[x]
S[2]：a[1] a[2] a[3] a[4]……a[x]
S[3]：a[1] a[2] a[3] a[4]……a[x]
……
S[x-1]：a[1] a[2] a[3]…a[x-1] a[x]
S[x]：a[1] a[2] a[3] a[4]……a[x]
这个方阵的和为(a[1]+a[2]+...+a[x])*(x+1)，而补上的蓝色部分的和为(a[1]*1+a[2]*2+...+xa[x])。
因此a[1-x]的前缀和为：(a[1]+a[2]+...+a[x])*(x+1)-(a[1]*1+a[2]*2+...+xa[x])==sum(a[i])*(x+1)-sum(a[i]*i)[1<=i<=x]。
所以，我们只需要维护a[i]和i*a[i]两个树状数组即可快速的完成这两个操作。

代码如下

#include <iostream>
#include <cstring>
#include <cstdio>
#define LL long long
using namespace std;
const int N=1e5+5;
int n,m;
int a[N];
LL tr1[N],tr2[N];
int lowbit(int x)		//树状数组的三个基本函数
{
    return x & -x;
}
void add(LL tr[],int x,LL c)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tr[i]+=c;
}
LL sum(LL tr[],int x)
{
    LL res=0;
    for(int i=x;i;i-=lowbit(i))
        res+=tr[i];
    return res;
}
LL query(int x)		//查询a[x]前缀和的前缀和SS[x]
{
    return sum(tr1,x)*(x+1)-sum(tr2,x);
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)	//建立树状数组
    {
        scanf("%d",&a[i]);
        add(tr1,i,a[i]);
        add(tr2,i,(LL)a[i]*i);
    }
    while(m--)
    {
        char op[10];
        int x,d;
        scanf("%s %d",op,&x);
        if(op[0]=='M')	//将a[i]修改为d，即在a[i]处加d-a[i]
        {
            scanf("%d",&d);
            add(tr1,x,d-a[x]);
            add(tr2,x,(LL)x*(d-a[x]));
            a[x]=d;			//修改原数组
        }
        else printf("%lld\n",query(x));		//查询操作
    }
    return 0;
}