P4114 Qtree1 （树剖）

题目链接

分析：
这题一开始以为就是个树剖板题，无非是把边权下放变成点权而已。
但写完发现wa了，原因是边权下放后 ，任意两点x y最短路上的点权是不包括LCA的点权的，因为下放给LCA的边权不在x y最短路径上。

所以链查询的时候要注意一下这里。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5+7;
const ll inf = 34359738370;
struct 
{
    int to,next,w;
}e[maxn<<1];
int head[maxn],num=0;
int de[maxn];//输入的第i条边的孩子节点
int c[maxn];//边权下放给孩子节点
void add(int u,int v,int w)
{
    e[++num].to=v;
    e[num].w=w;
    e[num].next=head[u];
    head[u]=num;
}
int n;
int size[maxn],fa[maxn],son[maxn],dep[maxn];
void dfs1(int rt,int par)
{
    size[rt]=1;
    for(int i=head[rt];i;i=e[i].next)
    {
        int to=e[i].to;
        if(to == par) continue;
        dep[to]=dep[rt]+1;
        fa[to]=rt;
        de[(i+1)/2]=to;//前向星的边从1开始计数 (i+1)/2就是这条边输入的序号
        c[to]=e[i].w;//边权下放
        dfs1(to,rt);
        size[rt]+=size[to];
        if(size[to] > size[son[rt]]) son[rt]=to;
    }
}
int pos[maxn],top[maxn],cnt=0,in[maxn];
//dfs序 链顶指针 
void dfs2(int rt,int t)
{
    in[rt]=++cnt;
    pos[cnt]=rt;
    top[rt]=t;
    if(son[rt]) dfs2(son[rt],t);
    for(int i=head[rt];i;i=e[i].next)
    {
        int to=e[i].to;
        if(to == fa[rt] || to==son[rt]) continue;
        dfs2(to,to); 
    }
}
//线段树维护dfs序区间最大值 单点修改
int tree[maxn<<2];
inline int lc(int &rt){ return rt<<1 ;}
inline int rc(int &rt){ return rt<<1|1 ;}
inline void pushup(int &rt){ tree[rt]=max(tree[lc(rt)],tree[rc(rt)]) ;}
inline void build(int rt,int l,int r)
{
    if(l == r)
    {
        tree[rt]=c[pos[l]];//pos数组中 下标i映射的是c[pos[i]]
        return ;
    }
    int mid=(l+r)>>1;
    build(lc(rt),l,mid);
    build(rc(rt),mid+1,r);
    pushup(rt);
}
inline void updata(int rt,int l,int r,int x,int v)
{
    if(l == r) 
    {
        tree[rt]=v;
        return ;
    }
    int mid=(l+r)>>1;
    if(x<=mid) updata(lc(rt),l,mid,x,v);
    else updata(rc(rt),mid+1,r,x,v);
    pushup(rt);
}
inline int query(int rt,int l,int r,int vl,int vr)
{
    if(l>vr || r<vl) return 0;
    if(vl<=l && r<=vr) return tree[rt];
    int mid=(l+r)>>1;
    return max(query(lc(rt),l,mid,vl,vr),query(rc(rt),mid+1,r,vl,vr));
}
inline int chainq(int x,int y)
{
    int ans=0;
    int fx=top[x],fy=top[y];
    while(fx != fy)
    {
        if(dep[fx]<dep[fy]) swap(x,y),swap(fx,fy);
        ans=max(ans,query(1,1,n,in[fx],in[x]));
        x=fa[fx];
        fx=top[x];
    }
    if(in[x]>in[y]) swap(x,y);//此时的x是原先x y的LCA 
    ans=max(ans,query(1,1,n,in[x]+1,in[y]));//注意这里是[in[x]+1,in[y]] LCA的点权不能计算进去
    return ans;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y,w;
        scanf("%d %d %d",&x,&y,&w);
        add(x,y,w),add(y,x,w);
    }
    dfs1(1,0);
    dfs2(1,1);
    build(1,1,n);
    char s[20];
    int a,b;
    while(1)
    {
        scanf("%s",s);
        if(s[0]=='D') break;
        scanf("%d %d",&a,&b);
        if(s[0]=='Q')
        {
            if(a==b) 
            {
                puts("0");
                continue;
            }
            printf("%d\n",chainq(a,b));
        }
        else updata(1,1,n,in[de[a]],b);
    }
    return 0;
}