CodeChef "Chef and Churus" 分块+树状数组

题意

给定$n$个数和$n$个区间和$f[i]=\sum_{k=l_i}^{r_i} a_k$，支持两种操作：
1. 将$a_i$变为$x$；
2. 求编号为$x$到$y$的所有$f$的和。

$n\leq 100000$

分析

对于每个$f_i$，可以直接使用树状数组求出。
所以我们可以使用分块，中间的直接用块的答案，边上的用树状数组。

首先我们进行预处理，记录第i块中每个编号的个数，这里用前缀和，并求出每个块的和。

对于修改操作1 x y，我们要维护树状数组和块状数组。
树状数组：直接维护。
块状数组：找到每个块中$x$的个数，这块的$sum$加上$x\times (y-a[x])$。
最后把$a[x]$赋成$y$。

对于查询操作2 x y。
中间的块我们直接用块状数组的答案，$O(\sqrt n)$。
边上的两块我们用树状数组暴力求所有的$f[i]$，$O(\sqrt n log n)$。

小结

我们的想法是这样的：
某些东西比较好求，那就对它分块，同时维护求的方法。

代码

#include <bits/stdc++.h>
using namespace std;

typedef long long Lint;
const int N=100010;
const int S=400;

int n,m;
int fl[N],fr[N],a[N];
Lint ta[N];
int cnt[S][N],unit,num; Lint sumb[S];

inline int Read(void)
{
    int s=0,f=1; char c=getchar();
    for (;!isdigit(c);c=getchar()) if (c=='-') f=-1;
    for (;isdigit(c);c=getchar()) s=s*10+c-'0';
    return s*f;
}

inline int lowbit(int i)
{
    return i&-i;
}

void Ins(int i,int add)
{
    for (;i<=n;i+=lowbit(i)) ta[i]+=add;
}

void Update(int x,int y)
{
    Ins(x,y-a[x]);
    for (int i=1;i<=num;i++) sumb[i]=sumb[i]+(Lint)cnt[i][x]*(y-a[x]);
    a[x]=y;
}

Lint GetSum(int i)
{
    Lint sum=0;
    for (;i;i-=lowbit(i)) sum+=ta[i];
    return sum;
}

Lint Query(int l,int r)
{
    int lBlock=(l-1)/unit+1,rBlock=(r-1)/unit+1; Lint sum=0;
    if (lBlock==rBlock)
        for (int i=l;i<=r;i++)
            sum=sum+GetSum(fr[i])-GetSum(fl[i]-1);
    else
    {
        for (int i=lBlock+1;i<=rBlock-1;i++) sum+=sumb[i];
        for (int i=l;i<=lBlock*unit;i++) sum=sum+GetSum(fr[i])-GetSum(fl[i]-1);
        for (int i=(rBlock-1)*unit+1;i<=r;i++) sum=sum+GetSum(fr[i])-GetSum(fl[i]-1);
    }
    return sum;
}

int main(void)
{
    n=Read();
    for (int i=1;i<=n;i++) a[i]=Read();
    for (int i=1;i<=n;i++) fl[i]=Read(),fr[i]=Read();

    for (int i=1;i<=n;i++) Ins(i,a[i]);

    int nowBlock=0;
    unit=(int)sqrt(n),num=unit+(unit*unit!=n);
    for (int i=1;i<=n;i++)
    {
        if (i%unit==1) nowBlock++;
        cnt[nowBlock][fl[i]]++;
        cnt[nowBlock][fr[i]+1]--;
    }
    for (int i=1;i<=num;i++)
        for (int j=1;j<=n;j++) 
        {
            cnt[i][j]+=cnt[i][j-1];
            sumb[i]=sumb[i]+(Lint)cnt[i][j]*a[j];
        }

    int k,x,y;
    m=Read();
    for (int i=1;i<=m;i++)
    {
        k=Read(),x=Read(),y=Read();
        if (k==1)
            Update(x,y);
        else printf("%lld\n",Query(x,y));
    }

    return 0;
}