【BZOJ】3211: 花神游历各国

题意

\(n\)个点，第\(i\)个点值为\(a_i\)。\(m\)个询问，每次询问\([l, r]\)内的和或者将\([l, r]\)的每个值改为自己的算术平方根。（\(n \le 100000, m \le 200000, 0 \le a_i \le 10^9\)）

分析

\(10^9\)开几次方就到\(1\)或者\(0\)了，所以对于已经到\(1\)或\(0\)的我们可以直接跳过。

题解

我们用一个并查集来维护"下一个"。即\(p_i\)表示从当前点开始不为\(1\)和\(0\)的数的下一个位置（包括自己）。用并查集合并即可。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100005;
int n, a[N], p[N];
ll c[N];
inline void add(int x, int s) {
    for(; x<=n; x+=x&-x) {
        c[x]+=s;
    }
}
inline ll sum(int x) {
    ll y=0;
    for(; x; x-=x&-x) {
        y+=c[x];
    }
    return y;
}
inline int find(int x) {
    return x==p[x]?x:p[x]=find(p[x]);
}
inline int getint() {
    char c=getchar();
    int x=0;
    for(; c<'0'||c>'9'; c=getchar());
    for(; c>='0'&&c<='9'; c=getchar()) {
        x=x*10+c-48;
    }
    return x;
}
int main() {
    n=getint();
    for(int i=1; i<=n; ++i) {
        a[i]=getint();
        add(i, a[i]);
        p[i]=i;
    }
    p[n+1]=n+1;
    int m=getint();
    while(m--) {
        int x, l, r;
        x=getint();
        l=getint();
        r=getint();
        if(x==1) {
            printf("%lld\n", sum(r)-sum(l-1));
        }
        else {
            for(int i=find(l); i<=r; i=find(i+1)) {
                add(i, -a[i]);
                a[i]=sqrt(a[i]);
                add(i, a[i]);
                if(a[i]<=1) {
                    p[i]=find(p[i+1]);
                }
            }
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/iwtwiioi/p/4985810.html