[bzoj3211]花神游历各国

Description

写一个数据结构兹瓷区间求和和区间开方。
n<=10^5,m<=2*10^5,ai<=10^9

Solution

区间开平方？暴力开呗。
反正10^9也不需要开多少次，只不过常数大了点罢了。
用线段树维护区间和，还有区间标记，表示这个区间是不是还有可以开方的数(1&0)。
当然，用树状数组维护和再用并查集来打标记也是可以的，而且常数更小。
不过，有一个迷之问题，为什么一定要把0给标记，不然会惨绝人寰地超时？
表示奥妙重重。

Code

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm> 
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 100005
using namespace std;
typedef long long ll;
struct note{ll v;bool bz;}t[N*3];
int n,m,z,x,y,a[N];
void back(int v,int x) {
    t[v].v=x;if (x==1||x==0) t[v].bz=1;
}
note merge(note y,note z) {
    note x;x.v=y.v+z.v;x.bz=y.bz&&z.bz;return x;
}
void build(int v,int l,int r) {
    if (l==r) {
        back(v,a[l]);
        return;
    }
    int m=(l+r)/2;
    build(v*2,l,m);build(v*2+1,m+1,r);
    t[v]=merge(t[v*2],t[v*2+1]);
}
void revise(int v,int l,int r) {
    if (t[v].bz) return;
    if (l==r) {
        a[l]=sqrt(a[l]);
        back(v,a[l]);
        return;
    }
    int m=(l+r)/2;
    revise(v*2,l,m);revise(v*2+1,m+1,r);
    t[v]=merge(t[v*2],t[v*2+1]);
}
void change(int v,int l,int r,int x,int y) {
    if (t[v].bz) return;
    if (l==x&&r==y) {revise(v,l,r);return;}
    int m=(l+r)/2;
    if (y<=m) change(v*2,l,m,x,y);
    else if (x>m) change(v*2+1,m+1,r,x,y);
    else change(v*2,l,m,x,m),change(v*2+1,m+1,r,m+1,y);
    t[v]=merge(t[v*2],t[v*2+1]);
}
ll find(int v,int l,int r,int x,int y) {
    if (l==x&&r==y) return t[v].v;
    int m=(l+r)/2;
    if (y<=m) return find(v*2,l,m,x,y);
    else if (x>m) return find(v*2+1,m+1,r,x,y);
    else return find(v*2,l,m,x,m)+find(v*2+1,m+1,r,m+1,y);
}
int main() {
    scanf("%d",&n);
    fo(i,1,n) scanf("%d",&a[i]);
    build(1,1,n);
    for(scanf("%d",&m);m;m--) {
        scanf("%d%d%d",&z,&x,&y);
        if (z==1) printf("%lld\n",find(1,1,n,x,y));
        else change(1,1,n,x,y);
    }
}