傳送門

樹狀數組模板題。註意優化，假設某個數的值已經是1了的話。那麽我們以後就不用對他進行操作了，這個能夠用並查集實現。

這道題還有個坑的地方，給出查詢區間端點的a,b，有可能a>b。

#include<cstdio>
#include<cmath>
#include<cctype>
#include<iostream>
using namespace std;
inline void GET(int &t){
    char c;
    t=0;
    do{c=getchar();}while(!isdigit(c));
    while(isdigit(c)){t=t*10+c-'0';c=getchar();}
}
inline void GET(long long &t){
    char c;
    t=0;
    do{c=getchar();}while(!isdigit(c));
    while(isdigit(c)){t=t*10+c-'0';c=getchar();}
}
long long tree[100005],c[100005];
int fa[100005];
int n,m;
int root(int x)
{
    if(fa[x]==x)
        return x;
    else
        return fa[x]=root(fa[x]);
}
int lowbit(long long x)
{
    return x&-x;
}
void updata(int pos,long long val)
{
    while(pos<=n)
    {
        tree[pos]+=val;
        pos+=lowbit(pos);
    }
}
long long getsum(int pos)
{
    long long sum=0;
    while(pos>0)
    {
        sum+=tree[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
int main()
{
    GET(n);
    for(int i=1;i<=n;i++)
    {
        GET(c[i]);
        updata(i,c[i]);
        fa[i]=i;
    }
    GET(m);
    int k,l,r;
    for(int i=1;i<=m;i++)
    {
        GET(k);
        GET(l);
        GET(r);
        if(r < l) swap(l,r);
        if(k==1)
            printf("%lld\n",getsum(r)-getsum(l-1));
        else
        {
            for(int j=root(l);j<=r;j=root(j+1))
            {
                if(j==0)
                    break;
                updata(j,-c[j]);
                c[j]=sqrt(c[j]);
                updata(j,c[j]);
                if(c[j]==1)
                    fa[j]=j+1;
            }
        }
    }
}

CodeVS2492 上帝造題的七分鐘2(樹狀數組+並查集)