Tyvj P1941 「Poetize4」上帝造题的七分钟2

背景

XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。

描述

"第一分钟，X说，要有数列，于是便给定了一个正整数数列。
第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。
第三分钟，k说，要能查询，于是便有了求一段数的和的操作。
第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。
第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。
第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。"
——《上帝造题的七分钟·第二部》
所以这个神圣的任务就交给你了。

输入格式

第一行一个整数n，代表数列中数的个数。
第二行n个正整数，表示初始状态下数列中的数。
第三行一个整数m，表示有m次操作。
接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。
UPD：注意数据中有可能l>r，所以遇到这种情况请交换l和r。

输出格式

对于询问操作，每行输出一个回答。

测试样例1

输入

10 
1 2 3 4 5 6 7 8 9 10 
5 
0 1 10 
1 1 10 
1 1 5 
0 5 8 
1 4 8

输出

19 
7 
6

备注

对于30%的数据，1<=n,m<=1000，数列中的数不超过32767。
对于100%的数据，1<=n,m<=100000，1<=l,r<=n，数列中的数大于0，且不超过1e12。
注意l有可能大于r，遇到这种情况请交换l,r。

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

树状数组+并查集~

（先膜各位神犇~）

神奇的思路，前一段时间暴力开平方，次数多了就会有数变为1，用并查集记录1的父子关系，改变时直接跳过1，所以不会超时~

另外输入的数组c一定要开成long long，否则会……T……（我怎么知道是为什么啊！）

#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
#define ll long long

int n,m,fa[1000001];
ll a[1000001],c[1000001];

int lowbit(int u)
{
	return u&(-u);
}

void add(int u,ll v)
{
	for(int i=u;i<=n;i+=lowbit(i)) a[i]+=v;
}

ll getnum(int u)
{
	ll ans=0;
	for(int i=u;i;i-=lowbit(i)) ans+=a[i];
	return ans;
}

int findd(int u)
{
	return fa[u]==u ? u:fa[u]=findd(fa[u]);
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%lld",&c[i]),add(i,c[i]),fa[i]=i;fa[n+1]=n+1;
	scanf("%d",&m);
	while(m--)
	{
		int k,l,r;
		scanf("%d%d%d",&k,&l,&r);
		if(l>r) swap(l,r);
		if(!k)
		{
			for(int i=findd(l);i<=r;i=findd(i+1))
			{
				ll z=(ll)sqrt(c[i]);
				add(i,z-c[i]);c[i]=z;
				if(c[i]==1) fa[i]=i+1;
			}
		}
		else printf("%lld\n",getnum(r)-getnum(l-1));
	}
	return 0; 
}