【BZOJ3211】【花神游历各国】&【BZOJ3038】上帝造题的七分钟2（树状数组+链表）

Description

Input

Output

每次x=1时，每行一个整数，表示这次旅行的开心度

Sample Input

4
1 100 5 5
5
1 1 2
2 1 2
1 1 2
2 2 3
1 1 4

Sample Output

101
11
11

HINT

对于100%的数据， n ≤ 100000，m≤200000 ,data[i]非负且小于10^9

题解：
题目大意为区间的开方与求和。但问题在于树状数组是不能做到区间开方的，所以好像应该用线段树做。那为什么又可以用树状数组呢？我们知道，一个数x只要开方loglogx次后取整就会变成1，也就是说int范围内的数开方10次以内就会变成1，再开方就没有意义了。所以我们用链表记录一个数的右边（包括自己）最近的不是1的数，只修改这些数就好了，均摊复杂度O(N)。

代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define N 100005
#define lb(x) (x&-x)
using namespace std;
ll read()
{
    ll x=0,f=1;
    char c=getchar();
    while(c<'0' || c>'9') {if(c=='-') f=-1;c=getchar();}
    while(c<='9' && c>='0') {x=x*10+c-'0';c=getchar();}
    return x*f;
}
ll n,m,fa[N],x,l,r;
ll v[N],t[N];
int find(int x)
{
    if(fa[x]==x || !fa[x]) return fa[x]=x;
    return fa[x]=find(fa[x]);
}
void add(int x,ll v)
{
    while(x<=n)
    {
        t[x]+=v;
        x+=lb(x);
    }
}
ll query(int x)
{
    ll ret=0;
    while(x)
    {
        ret+=t[x];
        x-=lb(x);
    }
    return ret;
}
void mdy(int l,int r)
{
    for(int i=l;i<=r;i=find(i+1))
    {
        ll t=sqrt(v[i]);
        add(i,t-v[i]);
        v[i]=t;
        if(v[i]<=1) fa[i]=find(i+1);
    }
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    {
        v[i]=read();
        if(v[i]<=1) fa[i]=i+1;
        add(i,v[i]);
    }
    m=read();
    for(int i=1;i<=m;i++)
    {
        x=read(),l=read(),r=read();
        if(l>r) swap(l,r);
        if(!x) mdy(l,r);
        else printf("%lld\n",query(r)-query(l-1));
    }
    return 0;
}