玲珑杯（树状数组）

题目链接：点击打开链接

代码：

#include
#define ll long long
using namespace std;
int n,m;
int a[100005],b[100005];
struct BIT
{
    ll t[100005];
    void add(int x,ll v)
    {
        for(; x<=n; x+=x&(-x))t[x]+=v;
    }
    ll get(int x)
    {
        ll ans=0;
        for(; x; x-=x&(-x))ans+=t[x];
        return ans;
    }
} t1,t2;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)scanf("%d",&a[i]);
    for(int i=1; i<=n; i++)b[i]=a[i]*a[i];
    for(int i=1; i<=n; i++)t1.add(i,a[i]),t2.add(i,b[i]);
    for(int i=1; i<=m; i++)
    {
        int opt,x,y;
        scanf("%d%d%d",&opt,&x,&y);
        if(opt==1)
        {
            t1.add(x,-a[x]);
            t2.add(x,-b[x]);
            a[x]=y;
            b[x]=y*y;
            t1.add(x,a[x]);
            t2.add(x,b[x]);
        }
        else
        {
            printf("%lld\n",(y-x+1)*(t2.get(y)-t2.get(x-1))-(t1.get(y)-t1.get(x-1))*(t1.get(y)-t1.get(x-1)));
        }
    }
    return 0;
}