上帝造题的七分钟2 / 花神游历各国

题目

自己上网搜

题解

发现每个数开方次数不超过六次
所以我们对于修改可以暴力更改（因为不能打标记）
查询仍然用上某种数据结构
但修改太暴力了
每个数开方到了小于等于 (1) 时
我们就不需要再更改它了（无意义）
为了高效地修改，我们要用某种方法快速过掉一堆不需要修改的数
没错，并查集！！
它的祖先定义为后面（包括自己）第一个开方有意义的数
那么当它开方后的值小于等于 (1) 时，直接让它的 (fa) 指向下一个位置
再让待修改指针指向它后面需要开方的位置

那么“某种数据结构”就用树状数组好了

(Code)

#include<cstdio>
#include<iostream>
#include<cmath>
#define lowbit (x & (-x))
using namespace std;
typedef long long LL;

const int N = 1e5 + 5;
int n , m , fa[N];
LL a[N] , c[N];

inline int find(int x){return fa[x] == x ? x : fa[x] = find(fa[x]);}

inline void add(int x , LL v){for(; x <= n; x += lowbit) c[x] += v;}

inline LL query(int x)
{
	LL res = 0;
	for(; x; x -= lowbit) res += c[x];
	return res;
}

int main()
{
	scanf("%d" , &n);
	for(register int i = 1; i <= n; i++) scanf("%lld" , a + i) , add(i , a[i]) , fa[i] = i;
	fa[n + 1] = n + 1;
	scanf("%d" , &m);
	int l , r , k;
	while (m--)
	{
		scanf("%d%d%d" , &k , &l , &r);
		if (r < l) swap(l , r);
		if (k == 1) printf("%lld
" , query(r) -  query(l - 1));
		else {
			while (l <= r)
			{
				LL t = (LL)sqrt(a[l]);
				add(l , t - a[l]) , a[l] = t;
				fa[l] = a[l] <= 1 ? l + 1 : l;
				l = fa[l] == l ? l + 1 : find(l);
			}
		}
	}
}