【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;
}