D - Tree(树剖+线段树维护区间开根号)

Problem

给定一颗大小为n的树和q此操作，op0:将一条树链上所有点的权值开根号；op1:求一条树链上所有点的权值和。

Solution

• 显然要用树剖将树映射到区间上，看到区间和时想用线段树维护，场上感觉线段树好像没法通过打标记维护区间开根号。
• 点权是小于等于1e9的，对于一个点来说最多开六次点权就会变为1，如果某个区间的区间和为r-l+1，就不需要对这个区间再进行开根号操作。因此可以维护一个没有pushdown的线段树，暴力修改到底开根号，效率上最差（全部单点开根号开到1）应该是一般线段树的六分之一。

Code

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
const int maxn = 1e5+5;
int n,q,last[maxn],cnt,dep[maxn],son[maxn],dfn[maxn],fa[maxn],sizes[maxn],lk[maxn],tot,tops[maxn];
long long tree[maxn*4],a[maxn],b[maxn],ans;
struct edge{
	int v,next;
}e[maxn*2];
inline void add(int u,int v)
{
	e[++cnt].v=v;
	e[cnt].next=last[u];
	last[u]=cnt;
}
void dfs1(int u,int f)
{
	
	sizes[u]=1;
	dep[u]=dep[f]+1;
	fa[u]=f;
	for(int i=last[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==f) continue;
		dfs1(v,u);
		sizes[u]+=sizes[v];
		if(sizes[v]>sizes[son[u]])
		son[u]=v;
	}
}
void dfs2(int u,int tp)
{
	tops[u]=tp;
	dfn[u]=++tot;
	a[tot]=u;
	if(son[u]) dfs2(son[u],tp);
	for(int i=last[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v!=son[u]&&v!=fa[u])
		dfs2(v,v);
	}
	lk[u]=tot;
}
void pushup(int x)
{
	tree[x]=tree[x<<1]+tree[x<<1|1];
}
void build(int x,int l,int r)
{
	if(l==r)
	{
		tree[x]=b[a[l]];
		return ;
	}
	int mid=(l+r)>>1;
	build(x<<1,l,mid);
	build(x<<1|1,mid+1,r);
	pushup(x);
}
void update(int x,int l,int r,int kl,int kr)
{
	if(l>kr||r<kl||tree[x]==r-l+1) return ;
	if(l==r)
	{
		tree[x]=sqrt(tree[x]);
		return ;
	}
	int mid=(l+r)>>1;
	if(mid>=kl)
	update(x<<1,l,mid,kl,kr);
	if(mid+1<=kr)
	update(x<<1|1,mid+1,r,kl,kr);
	pushup(x);
}
void query(int x,int l,int r,int kl,int kr)
{
    if(l>kr||r<kl) return;
    if(l>=kl&&r<=kr)
    {
        ans+=tree[x];
        return ;
    }
    int mid=(l+r)>>1;
    query(x<<1,l,mid,kl,kr);
    query(x<<1|1,mid+1,r,kl,kr);
}
void cal(int x,int y,int z)
{
    while(tops[x]!=tops[y])
    {
        if(dep[tops[x]]<dep[tops[y]]) swap(x,y);
        if(z==0)
        update(1,1,n,dfn[tops[x]],dfn[x]);
        else
        query(1,1,n,dfn[tops[x]],dfn[x]);
        x=fa[tops[x]];
    }
    if(dep[x]<dep[y]) swap(x,y);
    if(z==0)
    update(1,1,n,dfn[y],dfn[x]);
    else
    query(1,1,n,dfn[y],dfn[x]);
}
int main()
{
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)
	scanf("%lld",&b[i]);
	for(int i=1,u,v;i<=n-1;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	dfs1(1,0);
	dfs2(1,1);
	build(1,1,n);
	for(int i=1,op,x,y;i<=q;i++)
	{
		ans=0;
		scanf("%d%d%d",&op,&x,&y);
		cal(x,y,op);
		if(op)
		cout<<ans<<endl;
	}
}