树剖

树剖

只是想来放一波代码= =

单点修改+区间求和+区间最大值

区间修改就在线段树上加个Lazy操作就可以了

注意：代码里面，权值都是在点上的

#include<iostream> 
#include<cstdio> 
#define N 300000 
#define ll long long 
using namespace std; 
ll maxxx=0; 
ll V[N]; 
ll Last[N],Next[N],End[N],tot; 
ll Size[N],Fa[N],Dep[N],Son[N],Top[N],Dfn[N],Old[N],cnt; 
//all of the arrays use old ids 
struct node{ll l,r,v,Max;}tree[N]; 
void Find(ll u,ll uu,ll dep){ 
    Size[u]=1,Fa[u]=uu,Dep[u]=dep;
    for(ll i=Last[u];i;i=Next[i]){  
        ll v=End[i]; 
        if(v==Fa[u])continue;//Go Wrong 1
        Find(v,u,dep+1);  
        Size[u]+=Size[v]; 
        if(Size[v]>Size[Son[u]])Son[u]=v;
    } 
} 
void Connect(ll u,ll anc){ 
    Top[u]=anc,Dfn[u]=++cnt,Old[cnt]=u; 
    if(Son[u])Connect(Son[u],anc); 
    for(ll i=Last[u];i;i=Next[i]){ 
        ll v=End[i]; 
        if(v!=Son[u] && v!=Fa[u])Connect(v,v); 
    } 
} 
void Make(ll p,ll x,ll y){ 
    tree[p].l=x,tree[p].r=y; 
    if(x>=y){ 
        tree[p].v=V[Old[x]];//Go Wrong 2:tree[p].v=V[x];
        tree[p].Max=V[Old[x]];//Go Wrong 2
        return; 
    } 
    ll mid=(x+y)>>1; 
    Make(p<<1,x,mid),Make((p<<1)+1,mid+1,y); 
    tree[p].v=tree[p<<1].v+tree[(p<<1)+1].v; 
    tree[p].Max=max(tree[p<<1].Max,tree[(p<<1)+1].Max); 
} 
void Modify(ll p,ll u,ll t){ 
    if(tree[p].l==tree[p].r){ 
        tree[p].v=t,tree[p].Max=t; 
        return; 
    }
    int mid=(tree[p].l+tree[p].r)>>1;
    if(u<=mid)Modify(p<<1,u,t);
    else Modify((p<<1)+1,u,t); 
    tree[p].v=tree[p<<1].v+tree[(p<<1)+1].v; 
    tree[p].Max=max(tree[p<<1].Max,tree[(p<<1)+1].Max); 
} 
ll TreeSum(ll p,ll x,ll y){ 
    if(tree[p].r<x || tree[p].l>y)return 0; 
    if(tree[p].l>=x && tree[p].r<=y)return tree[p].v; 
    return TreeSum(p<<1,x,y)+TreeSum((p<<1)+1,x,y); 
} 
ll GetSum(ll x,ll y){ 
    ll Ans=0; 
    while(Top[x]!=Top[y]){ 
        if(Dep[Top[x]]<Dep[Top[y]])swap(x,y); 
        Ans+=TreeSum(1,Dfn[Top[x]],Dfn[x]); 
        x=Fa[Top[x]];//Go Wrong 3
    } 
    if(Dep[x]>Dep[y])swap(x,y); 
    Ans+=TreeSum(1,Dfn[x],Dfn[y]); 
    return Ans; 
} 
ll TreeMax(ll p,ll x,ll y){ 
    if(tree[p].l>y || tree[p].r<x)return -1e9; 
    if(tree[p].l>=x && tree[p].r<=y)return tree[p].Max; 
    return max(TreeMax(p<<1,x,y),TreeMax((p<<1)+1,x,y)); 
} 
ll GetMax(ll x,ll y){ 
    ll Ans=-1e9; 
    while(Top[x]!=Top[y]){ 
        if(Dep[Top[x]]<Dep[Top[y]])swap(x,y); 
        Ans=max(Ans,TreeMax(1,Dfn[Top[x]],Dfn[x])); 
        x=Fa[Top[x]];//Go Wrong 3
    } 
    if(Dep[x]>Dep[y])swap(x,y); 
    Ans=max(Ans,TreeMax(1,Dfn[x],Dfn[y])); 
    return Ans; 
} 
void Ins(ll x,ll y){ 
    End[++tot]=y; 
    Next[tot]=Last[x]; 
    Last[x]=tot; 
} 
int main(){ 
    ll n,q;scanf("%lld",&n); 
    for(ll i=1;i<n;i++){ 
        ll a,b;scanf("%lld%lld",&a,&b); 
        Ins(a,b),Ins(b,a); 
    } 
    for(ll i=1;i<=n;i++)scanf("%lld",&V[i]); 
    Find(1,0,1); 
    Connect(1,1);
    Make(1,1,n);
    scanf("%lld",&q); 
    for(ll i=1;i<=q;i++){ 
        char s[10];ll a,b; 
        scanf("%s%lld%lld",s,&a,&b); 
        if(s[1]=='H')Modify(1,Dfn[a],b);
        if(s[1]=='M')printf("%lld\n",GetMax(a,b)); 
        if(s[1]=='S')printf("%lld\n",GetSum(a,b));
    }
    return 0; 
}  

批注了Go Wrong的地方都是WA了之后才发现的错误之处。希望以后不会再在这些地方出错。

Something Goes Wrong !

Go Wrong 1

没有跳过v==father[u]的时候的情况。由于连的是无向边，所以在建立搜索树的时候必须判断。

Go Wrong 2

线段树里面每个点的编号都是搜索树里的新编号，不是题目里直接给的编号。在赋值的时候应该对应新编号。

Go Wrong 3

当操作完该条重链之后，x应该跳到这条链顶端的点的父亲节点，而不是跳到自己的父亲节点（这里指的都是搜索树里面的父子关系）。x=father[x]是标准的错误写法，保证又WA又TLE，一定是x=father[top[x]]。

LCA

void GetLCA(int x,int y){
    while(Top[x]!=Top[y]){
        if(Dep[Top[x]]<Dep[Top[y]])swap(x,y);
        x=Fa[Top[x]];
    }
    if(Dep[x]>Dep[y])swap(x,y);
    return x;
}