hdu 6547 Tree（树链剖分+势能线段树）

题意：树上每个节点都有一个值 ai，现在有 2 种操作：
1. 将一条链上的所有节点的值开根号向下取整；
2. 求一条链上值的和；

分析：区间开根号属于数值快速递降至稳定的函数，考虑暴力修改。共n个数，单点修改的时间复杂度为logn，对1e9开方至多只需5次，故总时间复杂度为O(nlogn*5+qlogn)。

代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=1e5+5;
int head[maxn],f[maxn],d[maxn],son[maxn],size[maxn],top[maxn];
int a[maxn];
int cnt;
int dfn;//dfs序新编号
int nid[maxn],oid[maxn];
int val[maxn<<2],flag[maxn<<2];
int n,q;
 
struct edge{
	int to,next;
}e[maxn<<1];
 
void add(int from,int to) 
{     
    e[++cnt]={to,head[from]};
    head[from]=cnt;
}
 
void dfs1(int x,int fath)//求出size[x],d[x],f[x],son[x]
{
	size[x]=1;//子树大小 
	d[x]=d[fath]+1;//深度 
	son[x]=0;//重儿子 
	f[x]=fath;//父亲 
	for(int i=head[x];i;i=e[i].next)
	{
		int to=e[i].to;
		if(to==fath) continue;
		dfs1(to,x);
		size[x]+=size[to];
		if(size[son[x]]<size[to]) son[x]=to;//从儿子中选出重儿子 
	}
}
 
void dfs2(int x,int topx)//求出top[x] 
{
	top[x]=topx;//x所在重链的顶端
	nid[x]=++dfn;
	oid[dfn]=x;
	if(son[x]!=0) dfs2(son[x],topx);//先dfs重儿子 
	for(int i=head[x];i;i=e[i].next)
	{
		if(e[i].to!=f[x]&&e[i].to!=son[x])//不等于父亲且不等于重儿子 
		{
			dfs2(e[i].to,e[i].to);//再dfs轻儿子 
		}
	}
}
 
void pushup(int rt)
{
    val[rt]=val[rt<<1]+val[rt<<1|1];
    flag[rt]=flag[rt<<1]&flag[rt<<1|1];
}
 
void build(int l,int r,int rt)
{
    if(l==r)
	{
        val[rt]=a[oid[l]];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
 
void update(int L,int R,int l,int r,int rt)
{
    if(flag[rt]) return;
    if(l==r)
	{
        val[rt]=(int)sqrt(val[rt]);
        if(val[rt]<=1) flag[rt]=1;
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid) update(L,R,l,mid,rt<<1);
    if(R>mid) update(L,R,mid+1,r,rt<<1|1);
    pushup(rt);
}
 
int query(int L,int R,int l=1,int r=n,int rt=1)
{
	int res=0;
    if(L<=l&&r<=R) return val[rt];
    int m=(l+r)>>1;
    if(L<=m) res+=query(L,R,l,m,rt<<1);
	if(m<R) res+=query(L,R,m+1,r,rt<<1|1);
    return res;
}
 
void change(int x,int y)
{
	while(top[x]!=top[y])//不在一条重链上 
	{
		if(d[top[x]]<d[top[y]]) swap(x,y);
		update(nid[top[x]],nid[x],1,n,1);
		x=f[top[x]];//使深的那个点(x)跳到top[x] 
	}
	if(d[x]>d[y]) swap(x,y);
	update(nid[x],nid[y],1,n,1);
}
 
int qquery(int x,int y)
{
	int res=0;
	while(top[x]!=top[y])//不在一条重链上 
	{
		if(d[top[x]]<d[top[y]]) swap(x,y);
		res+=query(nid[top[x]],nid[x],1,n,1);
		x=f[top[x]];//使深的那个点(x)跳到top[x] 
	}
	if(d[x]>d[y]) swap(x,y);
	res+=query(nid[x],nid[y],1,n,1);
	return res;
}
 
signed main()
{
//	cnt=dfn=0;
    scanf("%lld%lld",&n,&q);
//    for(int i=1;i<=n;i++) head[i]=0;
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n-1;i++)
    {
        int u,v;
        scanf("%lld%lld",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs1(1,0);
    dfs2(1,1);
    build(1,n,1);
    while(q--)
    {
        int op,u,v;
        scanf("%lld",&op);
        if(op==0)
        {
            scanf("%lld%lld",&u,&v);
            change(u,v);
        }
        else
		{
			scanf("%lld%lld",&u,&v);
			printf("%lld\n",qquery(u,v));
		}
    }
}