1554 异象石

1554 异象石

题目说的，给定n个点n-1个边，那么肯定是一个树形结构
当然，我是来练习LCA的，因为我感觉练习LCA有点少了
地图上某一个点出现了异象石，地图上某个点的异象石被摧毁，向玩家询问所有异象石所在的联通点的的总长度的长度最小是多少
可以看出来了是一个求LCA，这里不成一个常识，菊花图，表示任意一个点都和根节点有一条连边
我们可以先对这棵树进行深度优先遍历，求出时间戳
仔细考虑，瞎猜找规律就发现，如果我们按照时间戳从小到大的顺序，把出现异象石的节点排成一圈，并且累加相邻两个节点的路径长度，把出现异象石的节点排成一圈之后，并且累加相邻两个节点之间的路径长度，最后得到的结果恰好就是结果的两倍

#include<bits/stdc++.h>
using namespace std;
 
typedef long long LL;
#define lowbit(x) (x)&(-x)
#define iter set<int>::iterator
const int maxn=1e5+10,maxk=1e4+10,MOD=10056;
LL n,m,pos[maxn],dep[maxn],dfn[maxn],f[maxn][25],d[maxn];
LL ans,cnt;
LL nxt[2*maxn],to[2*maxn],he[2*maxn],w[2*maxn],en;
vector<int> node[maxn];
set<int> seq;
//倍增法求LCA
struct edge{
    int to,dis;
    edge(int t,int d):to(t),dis(d){}
};
vector<edge> edges;
struct node{
    int num,front,next;
    node(int nu,int fr,int ne):num(nu),front(fr),next(ne){}
    bool operator < (node& x){
        return num<x.num;
    }
};
inline void add(int x,int y,int dis){
    to[++en]=y,w[en]=dis,nxt[en]=he[x],he[x]=en;
}
void dfs(int u,int father){
    dep[u]=dep[father]+1;
    dfn[u]=++cnt,pos[cnt]=u;
    for (int i=0;i<=19;i++)
        f[u][i+1]=f[f[u][i]][i];
    for (int e=he[u];e;e=nxt[e]){
        int v=to[e];
        if (v==father) continue;
        f[v][0]=u;
        d[v]=d[u]+w[e];
        dfs(v,u);
    }
}
int LCA(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=20;i>=0;i--) if (dep[f[x][i]]>=dep[y]) x=f[x][i]; //把x调到与y同一层
    if (x==y) return x;
    for (int i=20;i>=0;i--) if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; //不同才跳
    return f[x][0];
}
inline LL path(int x,int y){ return d[x]+d[y]-2*d[LCA(x,y)];}
iter left(iter x){
    if (x==seq.begin()) return --seq.end();
    return --x;
}
iter right(iter x){
    if (x==--seq.end()) return seq.begin();
    return ++x;
}
int main() {
    scanf("%d",&n);
    for (int i=0;i<n-1;i++){
        int x,y,dis;
        scanf("%d%d%d",&x,&y,&dis);
        add(x,y,dis),add(y,x,dis);
    }
    dep[0]=0,f[1][0]=0,d[1]=0;
    dfs(1,0);
    scanf("%d",&m);
    for (int i=0;i<m;i++){
        char ch;
        scanf(" %c",&ch);
        if (ch=='?') printf("%lld\n",ans/2);
        else if (ch=='+'){
            int num;
            scanf("%d",&num);
            if (seq.empty()) seq.insert(dfn[num]);
            else {
                iter r=seq.lower_bound(dfn[num]);
                if (r==seq.end()) r=seq.begin();
                iter l=left(r);
                ans+=path(pos[*l],num)+path(num,pos[*r])-path(pos[*l],pos[*r]);
                seq.insert(dfn[num]);
            }
        }
        else {
            int num;
            scanf("%d",&num);
            iter it=seq.find(dfn[num]);
            iter l=left(it),r=right(it);
            ans+=path(pos[*l],pos[*r])-path(pos[*l],num)-path(num,pos[*r]);
            seq.erase(dfn[num]);
        }
    }
}