题意
(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
", 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;
}