lca:异象石（set+dfs序）

题目：https://loj.ac/problem/10132

#include<bits/stdc++.h>
using namespace std;
int n,m,tot=,N,k=,head[];
struct node{
    int to,next,w;
}e[];
int depth[],grand[][],c[],num[];
long long d[];
void add(int x,int y,int c)
{
    e[++tot].next=head[x],e[tot].w=c,e[tot].to=y,head[x]=tot;
}
void init()
{
    N=floor(log(n+0.0)/log(2.0));
    depth[]=;
}
void dfs(int x)
{
    num[x]=++k;c[k]=x;
    for(int i=;i<=N;i++)grand[x][i]=grand[grand[x][i-]][i-];
    for(int i=head[x];i;i=e[i].next)
    {
        int v=e[i].to;
        if(v==grand[x][])continue;
        depth[v]=depth[x]+;
        grand[v][]=x;
        d[v]=d[x]+e[i].w;
        dfs(v);
    }
}
int lca(int a,int b)
{
    if(depth[a]>depth[b]) swap(a,b);
    for(int i=N;i>=;i--){
        if(depth[a]<depth[b]&&depth[grand[b][i]]>=depth[a]) b=grand[b][i];
    }
    if(a==b) return a;
    for(int i=N;i>=;i--){
        if(grand[a][i]!=grand[b][i]){a=grand[a][i],b=grand[b][i];}
    }
    return grand[a][];
}
set<int> s;
#define Auto set<int>::iterator
Auto Left(Auto t)
{
    if(t==s.begin()) return --s.end();
    return --t;
}
Auto Right(Auto t)
{
    if(t==--s.end()) return s.begin();
    return ++t;
}
long long dist(int x,int y)
{
    return d[x]+d[y]-(d[lca(x,y)]<<);
}
int main()
{
    char ch[];
    scanf("%d",&n);
    for(int i=;i<n;i++)
    {
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        add(x,y,z),add(y,x,z);
    }
    init();
    dfs();
    scanf("%d",&m);
    Auto l,r;
    long long ans=;
    for(int i=;i<=m;i++){
        scanf("%s",ch);
        if(ch[]=='?') printf("%lld\n",ans>>);
        else{
            int x;
            scanf("%d",&x);
            if(ch[]=='+')
            {
                if(!s.empty())
                {
                    r=s.lower_bound(num[x]);
                    if(r==s.end())r=s.begin();
                    l=Left(r);
                    ans+=dist(c[*l],x)+dist(x,c[*r])-dist(c[*l],c[*r]);
                }
                s.insert(num[x]);
            }
            else{
                r=s.find(num[x]);
                l=Left(r);r=Right(r);
                ans-=dist(c[*l],x)+dist(x,c[*r])-dist(c[*l],c[*r]);
                s.erase(num[x]);
            }
        }
    }
    return ;
}