[CC]Chef and Churu

题意简述

给你$n$个数，$n$个函数。
每个函数的意义是数$[L,R]$的和。
$q$次操作，每次询问一段函数的和，或更改一个数。

数据范围

$1 \leq n,q \leq 10^5$
$1 \leq a_i \leq 10^9$

思路

对函数分块。
每块建立一个系数数组，表示有多少个$a_i$。
每块维护一个答案。
这样我们就可以$O(\sqrt n)$完成整块的维护和查询了。
那么散点怎么办？
我们对数在分块。
维护一个前缀和，每块维护一个加标记。
每次修改块内重构，块外打标记。
复杂度$O(q\sqrt n)$

代码

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define block 310
#define MAXN 100010
typedef unsigned long long ll;
int n,u,v,q,opt,bu,bv;
int l[MAXN],r[MAXN],xi[400][MAXN],a[MAXN];
ll seq[MAXN],add[400],f[400],ans,w;
ll calc(int x)
{
    ll ret=seq[r[x]]+add[r[x]/block];
    if (l[x])
        ret-=seq[l[x]-1]+add[(l[x]-1)/block];
    return ret;
}
int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;i++)
        scanf("%llu",&seq[i]);
    for (int i=0;i<n;i++)
    {
        scanf("%d%d",&l[i],&r[i]);
        l[i]--,r[i]--;
    }
    for (int i=0;i*block<n;i++)
    {
        memset(a,0,sizeof(a));
        for (int j=i*block;j<min((i+1)*block,n);j++)
            a[l[j]]++,a[r[j]+1]--;
        for (int j=1;j<n;j++)
            a[j]+=a[j-1];
        for (int j=0;j<n;j++)
        {
            xi[i][j]=a[j];
            f[i]+=xi[i][j]*seq[j];
        }
    }
    for (int i=1;i<n;i++)
        seq[i]+=seq[i-1];
    scanf("%d",&q);
    for (int i=0;i<q;i++)
    {
        scanf("%d",&opt);
        if (opt==1)
        {
            scanf("%d%llu",&u,&w);
            u--;
            bu=u/block;
            w-=((seq[u]+add[bu])-(u!=0)*(seq[u-1]+add[(u-1)/block]));
            for (int j=bu*block;j<u;j++)
                seq[j]+=add[bu];
            for (int j=u;j<min((bu+1)*block,n);j++)
                seq[j]+=w+add[bu];
            add[bu]=0;
            for (int j=bu+1;j*block<n;j++)
                add[j]+=w;
            for (int j=0;j*block<n;j++)
                f[j]+=xi[j][u]*w;
        }
        if (opt==2)
        {
            scanf("%d%d",&u,&v);
            u--,v--;
            bu=u/block,bv=v/block;
            ans=0;
            if (bu==bv)
                for (int j=u;j<=v;j++)
                    ans+=calc(j);
            else
            {
                for (int j=u;j<min((bu+1)*block,n);j++)
                    ans+=calc(j);
                for (int j=bv*block;j<=v;j++)
                    ans+=calc(j);
                for (int j=bu+1;j<bv;j++)
                    ans+=f[j];
            }
            printf("%llu\n",ans);
        }
    }
    return 0;
}