首页 > 代码库 > BZOJ 3038 上帝造题的七分钟2 (并查集+树状数组)

BZOJ 3038 上帝造题的七分钟2 (并查集+树状数组)

题解:同 BZOJ 3211 花神游历各国需要注意的是需要开long long,还有左右节点需要注意一下。

#include <cstdio>#include <cmath>#include <iostream>#include <algorithm>using namespace std;typedef long long LL;LL a[100005],c[100005];int f[100005],n,m,op,l,r,t;int sf(int x){return x==f[x]?x:f[x]=sf(f[x]);}int update(int x,LL num){while(x<=n)c[x]+=num,x+=x&-x;}LL sum(int x){LL s=0;while(x)s+=c[x],x-=x&-x;return s;}int main(){    scanf("%d",&n);    for(int i=1;i<=n;i++)f[i]=i,scanf("%lld",&a[i]),update(i,a[i]);    scanf("%d",&m); f[n+1]=n+1;    for(int i=1;i<=m;i++){        scanf("%d%d%d",&op,&l,&r);        if(l>r)swap(l,r);        if(op==1)printf("%lld\n",(sum(r)-sum(l-1)));        else for(int j=l;j<=r;update(j,(t=(int)sqrt(a[j]))-a[j]),a[j]=t,        f[j]=(a[j]<=1)?j+1:j,j=(sf(j)==j?j+1:f[j]));    }    return 0;    }