[BZOJ3211]花神游历各国&&[BZOJ3038] 上帝造题的七分钟2 树状数组+并查集

3211: 花神游历各国

Time Limit: 5 Sec  Memory Limit: 128 MB
Submit: 4057  Solved: 1480
[Submit][Status][Discuss]

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

 

 

Source

SPOJ2713 gss4 数据已加强

 

由于是开方，所以每个数的开放次数不超过logn次，用树状数组维护前缀和，对于修改操作变为单点修改。

用并查集维护当前节点之前的最近的一个非1节点。

复杂度o（nlog²n）

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int read() {
    int x=0;char ch=getchar();
    while(!isdigit(ch)) ch=getchar();
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x;
}
int n,m;
long long sum[100005];
long long a[100005];
int fa[100005];
int find(int x){return fa[x]==x?fa[x]:fa[x]=find(fa[x]);}
int lowbit(int x){return x&(-x);}
void add(int x,long long val) {
    for(int i=x;i<=n;i+=lowbit(i)) sum[i]+=val;
}
void change(int x,long long val) {
    for(int i=x;i<=n;i+=lowbit(i)) sum[i]=sum[i]-val+(long long)sqrt(val);
}
long long query(int x) {
    long long ans=0; 
    for(int i=x;i>0;i-=lowbit(i)) ans+=sum[i];
    return ans;
}
int main() {
    n=read();
    for(int i=1;i<=n;i++) {a[i]=read();add(i,a[i]);}
    for(int i=1;i<=n;i++) fa[i]=i;
    m=read();
    while(m--) {
        int x=read(),l=read(),r=read();
        if(x==2) {
            int now=find(r);
            while(now>=l) {
                change(now,a[now]);
                a[now]=sqrt(a[now]);
                if(a[now]<=1) fa[now]=find(fa[now-1]);
                now=find(fa[now-1]);
            }
        }
        else {
            printf("%lld\n",query(r)-query(l-1));
        }
    }
    return 0;
}