【BZOJ 3155】Preprefix sum（树状数组）

最新推荐文章于 2019-08-07 23:11:13 发布

Patrickpwq

最新推荐文章于 2019-08-07 23:11:13 发布

阅读量130

点赞数

分类专栏： Patrickpwq的数据结构专栏

本文链接：https://blog.csdn.net/Patrickpwq/article/details/83422744

版权

Patrickpwq的数据结构专栏专栏收录该内容

35 篇文章 2 订阅

订阅专栏

博客介绍了如何利用树状数组巧妙地处理前缀和的前缀和问题，强调了直接使用两个树状数组的弊端，并提出了优化方案。

摘要由CSDN通过智能技术生成

显然是不能直接开两个树状数组维护前缀和，前缀和的前缀和。因为一旦对a[i]进行修改，将会影响许多位前缀和的前缀和

我们考虑对式子变一下形

Qi =S1+S2+S3+...+Si

    =a1+a1+a2+a1+a2+a3+...+ai

    =a1*i+a2*(i-1)+a3*(i-2)+...+ai*

    =(a1+a2+a3+...+ai)*i - (a2+a3^2+a4^3+...+ai^(i-1))

因此还是开两个树状数组一个维护前半部分一个维护后半部分

#include<bits/stdc++.h>
#define N 100005
using namespace std;
typedef long long ll;
struct BIT
{
    ll tree[N];
    int n;
    inline int lowbit(int x)
    {
        return x&(-x);
    }
    inline void update(int x,ll del)
    {
        for(int i=x;i<=n;i+=lowbit(i))  tree[i]+=del;
    }
    inline ll query(int x)
    {
        ll ans=0;
        for(int i=x;i;i-=lowbit(i)) ans+=tree[i];
        return ans;
    }
}tree1,tree2;
template<class T>
inline void read(T &x)
{
    x=0;
    static char ch=getchar();
    while(ch<'0'||ch>'9')   ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
int n,m;
ll a[N];
int main()
{
    read(n); read(m);
    tree1.n=n;  tree2.n=n;
    for(int i=1;i<=n;i++)
    {
        read(a[i]);
        tree1.update(i,a[i]);
        tree2.update(i,a[i]*(i-1));
    }
    char s[10];
    for(int i=1;i<=m;i++)
    {
        scanf("%s",&s);
        ll x,y;
        if(s[0]=='Q')
        {
            read(x);
            cout<<ll(tree1.query(x)*x)-tree2.query(x)<<'\n';
        }
        else
        {
            read(x);    read(y);
            tree1.update(x,y-a[x]); tree2.update(x,(y-a[x])*(x-1));
            a[x]=y;
        }
    }
    return 0;
}