【BZOJ】3211: 花神游历各国

时间:2022-01-08 04:15:22

题意

\(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;
}