2019ccpc女生專場 D - Tree

http://acm.hdu.edu.cn/showproblem.php?pid=6547
題意：給定一棵樹，有兩種操作，路徑點權和查詢，路徑上每個點開根號向下取整。
思路：明顯是樹剖的操作，但是這個開根號沒辦法維護，線段樹部分複雜度爆炸。但是實際上，每個值都在1e9之內，開根號不到10次就到1了，那麼，我們只需要開一個bool變量維護一段區間是不是全都是1就可以了。

#include<bits/stdc++.h>
using namespace std;
#define maxn (100000+100)
#define ll long long

int n,m,root=1,dfs_clock,op;
vector<int> G[maxn];
int fa[maxn],son[maxn],sz[maxn],deep[maxn],id[maxn],id2[maxn],top[maxn];
ll w[maxn],ver_w[maxn];

int ql,qr;
ll sumv[maxn*4],all_one[maxn*4];
ll _sum;

ll build(int o,int l,int r)
{
    int mid=(l+r)/2;
    if(l==r)return sumv[o]=w[l];
    return sumv[o]=build(o*2,l,mid)+build(o*2+1,mid+1,r);
}

void maintain(int o,int l,int r)
{
    if(l==r)
	{
		sumv[o]=w[l];
		all_one[o]=(w[l]==1);
	}
    else
	{
		sumv[o]=sumv[o*2]+sumv[o*2+1];
		all_one[o]=(all_one[o*2]&&all_one[o*2+1]);
	}   
}

void query(int o,int l,int r)
{
    if(ql<=l&&qr>=r)_sum+=sumv[o];
    else
    {
        int mid=(l+r)/2;
        if(ql<=mid)query(o*2,l,mid);
        if(qr>mid)query(o*2+1,mid+1,r);
    }  
}

void update(int o,int l,int r)
{
    if(all_one[o])return;
	else if(l==r)w[l]=sqrt(w[l]);
    else
    {
        int mid=(l+r)/2;
        if(ql<=mid)update(o*2,l,mid);
        if(qr>mid)update(o*2+1,mid+1,r);
    }
    maintain(o,l,r);
}

void dfs1(int u,int f)
{
    deep[u]=deep[f]+1;
    sz[u]=1;
    fa[u]=f;
    int maxx=0;
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==f)continue;
        dfs1(v,u);
        sz[u]+=sz[v];
        if(sz[v]>maxx){maxx=sz[v];son[u]=v;}
    }
}

void dfs2(int u,int up)
{
    id[u]=id2[u]=++dfs_clock;
    top[u]=up;
    if(son[u]){dfs2(son[u],up);id2[u]=id2[son[u]];}
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa[u] || v==son[u])continue;
        dfs2(v,v);
        id2[u]=id2[v];
    }
}

void Update1(int u,int v)
{
    int tpu=top[u],tpv=top[v];
    while(tpu!=tpv)
    {
        if(deep[tpu]<deep[tpv]){swap(tpu,tpv);swap(u,v);}
        ql=id[tpu],qr=id[u];
        update(1,1,n);
        u=fa[tpu];
        tpu=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    ql=id[u],qr=id[v];
    update(1,1,n);
}

ll Query1(int u,int v)
{
    _sum=0;
    int tpu=top[u],tpv=top[v];
    while(tpu!=tpv)
    {
        if(deep[tpu]<deep[tpv]){swap(tpu,tpv);swap(u,v);}
        ql=id[tpu],qr=id[u];
        query(1,1,n);
        u=fa[tpu];
        tpu=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    ql=id[u],qr=id[v];
    query(1,1,n);
    return _sum;
}

int main()
{
    //freopen("input.in","r",stdin);
    while(cin>>n>>m)
	{
		dfs_clock=0;
		memset(sumv,0,sizeof(sumv));
		memset(all_one,0,sizeof(all_one));
		memset(son,0,sizeof(son));
		int x,y,z;
		for(int i=1;i<=n;i++)scanf("%d",&ver_w[i]),G[i].clear();
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&x,&y);
			G[x].push_back(y);
			G[y].push_back(x);
		} 
		dfs1(root,0);
		dfs2(root,root);
		for(int i=1;i<=n;i++)w[id[i]]=ver_w[i];
		build(1,1,n);  
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&op,&x,&y);
			switch(op)
			{
				case 0:
					Update1(x,y);
					break;
				case 1:
					printf("%lld\n",Query1(x,y));
					break;
			}
		}
	}
    return 0;
}