首页 > 代码库 > BZOJ 3211 花神游历各国 (树状数组+并查集)
BZOJ 3211 花神游历各国 (树状数组+并查集)
题解:首先,单点修改求区间和可以用树状数组实现,因为开平方很耗时间,所以在这个方面可以优化,我们知道,开平方开几次之后数字就会等于1 ,所以,用数组记录下一个应该开的数,每次直接跳到下一个不是1的数字进行开平方,至于这个数组,可以用并查集维护。
#include <cstdio>#include <cmath>#include <iostream>using namespace std;typedef long long LL;LL c[100005];int a[100005],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,int 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("%d",&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(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; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。