Tree【HDU 6547】【女赛补题 树剖】

题目链接

Problem Description
wls 有三棵树，树上每个节点都有一个值 ai，现在有 2 种操作：
1, 将一条链上的所有节点的值开根号向下取整；
2 , 求一条链上值的和；
链的定义是两点之间的最短路。

Input
第一行两个数 n, q 分别代表树上点的数量和操作数量。
第二行 n 个整数，第 i 个数代表第 i 个点的值 ai。
接下来 n − 1 行， 每行两个整数 u, v 代表 u，v 之间有一条边。数据保证点两两联通。
接下来 q 行，每行有个整数 op, u, v，op = 0 表示将 u, v 这条链上所有的点的值开根号向下取整，op = 1表示询问 u，v 这条链上的值的和。
1 ≤ n, q ≤ 100, 000
0 ≤ ai ≤ 1, 000, 000, 000

Output
对于每一组 op = 2 的询问，输出一行一个值表示答案。

解题思路

当时现场赛的时候一眼就看出来是个树剖，奈何没有学过树剖，照着板子敲了一会实在是懵逼。。。然后也没写出来。。。
起那两天重现赛把题放出来了，刚好也学了树剖，就补一下。
中间可以写一个剪枝，当这条路上所有的点权都为1的时候，就不用再继续开根号了，也就是e[root].sum==(e[root].r-e[root].l+1)的时候。
好像不剪枝也是可以过的。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;

const int N=1e5+7;
int fi[N],tot;
long long a[N];
int n,m,f[N],d[N],siz[N],son[N],rk[N],top[N],id[N],cnt;
struct node
{
    int v,nex;
};
node edge[N*2];
struct Node
{
    int l,r;
    long long sum;
};
Node e[N*4];

void add(int u,int v)
{
    edge[tot].v=v;
    edge[tot].nex=fi[u];
    fi[u]=tot++;
}
void dfs1(int u)
{
    siz[u]=1;
    d[u]=d[f[u]]+1;
    for(int i=fi[u];i!=-1;i=edge[i].nex)
    {
        int v=edge[i].v;
        if(v!=f[u])
        {
            f[v]=u;
            dfs1(v);
            siz[u]+=siz[v];
            if(siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}
void dfs2(int u,int topp)
{
    top[u]=topp;
    id[u]=++cnt;
    rk[cnt]=u;
    if(son[u])
        dfs2(son[u],topp);
    for(int i=fi[u];i!=-1;i=edge[i].nex)
    {
        int v=edge[i].v;
        if(v!=f[u]&&v!=son[u])
            dfs2(v,v);
    }
}

//线段树
void build(int root,int l,int r)
{
    e[root].l=l;
    e[root].r=r;
    if(e[root].l==e[root].r)
    {
        e[root].sum=a[rk[l]];
        return ;
    }
    else
    {
        build(root*2,l,(l+r)/2);
        build(root*2+1,(l+r)/2+1,r);
        e[root].sum=e[root*2].sum+e[root*2+1].sum;
    }
}
void update(int root,int l,int r)
{
    if(e[root].sum==(e[root].r-e[root].l+1))
        return ;
    if(e[root].l==e[root].r)
    {
        e[root].sum=(long long)sqrt(e[root].sum);
        return ;
    }
    else
    {
        int mid=(e[root].l+e[root].r)/2;
        if(l<=mid)
            update(root*2,l,r);
        if(r>mid)
            update(root*2+1,l,r);
        e[root].sum=e[root*2].sum+e[root*2+1].sum;
    }
}
void updates(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(d[top[x]]<d[top[y]])
            swap(x,y);
        update(1,id[top[x]],id[x]);
        x=f[top[x]];
    }
    if(d[x]>d[y])
        swap(x,y);
    update(1,id[x],id[y]);
}
long long que(int root,int l,int r)
{
    if(e[root].l>=l&&e[root].r<=r)
        return e[root].sum;
    int mid=(e[root].l+e[root].r)/2;
    long long ans=0;
    if(l<=mid)
        ans+=que(root*2,l,r);
    if(r>mid)
        ans+=que(root*2+1,l,r);
    return ans;
}
long long ques(int x,int y)
{
    long long ans=0;
    while(top[x]!=top[y])
    {
        if(d[top[x]]<d[top[y]])
            swap(x,y);
        ans+=que(1,id[top[x]],id[x]);
        x=f[top[x]];
    }
    if(d[x]>d[y])
        swap(x,y);
    ans+=que(1,id[x],id[y]);
    return ans;
}
int main()
{
    memset(fi,-1,sizeof(fi));
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    int x,y,z;
    for(int i=1;i<n;i++)
    {
        scanf("%d %d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs1(1);
    dfs2(1,1);
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d %d",&z,&x,&y);
        if(z==0)
            updates(x,y);
        else
            printf("%lld\n",ques(x,y));
    }
    return 0;
}