序列 ssl 2650 线段树+并查集

题目大意

有一个n个数的序列A，请写一个程序完成下列操作：
1 l r表示查询l到r的和
2 l r表示把每个a[x] (l<=x<=r)变为sqrt(x)
n<=100000,m<=100000

分析

显然开根号是不能区间维护的，但发现一个int最多开五次根号就会变为1，所以我们可以用并查集来维护每个数的下一个可以开根号的数是哪一个，那么就可以均摊O(1)来暴力修改了。
区间操作用线段树搞定即可。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>

using namespace std;

struct arr{
    int x,y;
    double sum;
}f[3000000];
int a[200000];
int next[200000];
int n,m;

int insert(int r,int x,int y,double add)
{
    f[r].sum+=(y-x+1)*add;
    if ((f[r].x==x)&&(f[r].y==y)) 
        return 0;
    int mid=(f[r].x+f[r].y)/2;
    if (y<=mid) insert(r*2,x,y,add);
        else if (x>mid) insert(r*2+1,x,y,add);
            else insert(r*2,x,mid,add),insert(r*2+1,mid+1,y,add);
}

double find(int r,int x,int y)
{
    if ((f[r].x==x)&&(f[r].y==y)) return f[r].sum;
    int mid=(f[r].x+f[r].y)/2;
    if (y<=mid) return find(r*2,x,y);
        else if (x>mid) return find(r*2+1,x,y);
            else return find(r*2,x,mid)+find(r*2+1,mid+1,y);
}

int maketree(int r,int x,int y)
{
    f[r].x=x; f[r].y=y; 
    f[r].sum=0;
    if (x==y) return 0;
    int mid=(x+y)/2;
    maketree(r*2,x,mid),maketree(r*2+1,mid+1,y);
}

int find(int x)
{
    if (next[x]==x) return x;
    next[x]=find(next[x]);
    return next[x];
}

int main()
{
    freopen("sequence.in", "r", stdin);
    freopen("sequence.out", "w", stdout);
    scanf("%d",&n);
    maketree(1,1,n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        insert(1,i,i,a[i]);
    }
    scanf("%d",&m);
    for (int i=1;i<=n+1;i++)
        next[i]=i;
    for (int i=1;i<=n;i++)
        if (a[i]<=1) next[find(i)]=find(i+1);
    for (int i=1;i<=m;i++)
    {
        int w,x,y;
        scanf("%d%d%d",&w,&x,&y);
        if (w==1)
            printf("%.0f\n",find(1,x,y));
        else
        {
            if (a[x]<=1) x=find(x+1);
            while (x<=y)
            {
                insert(1,x,x,-a[x]);
                a[x]=sqrt(a[x]);
                insert(1,x,x,a[x]);
                if (a[x]==1)    next[find(x)]=find(x+1);
                x=find(x+1);
            }
        }
    }
}