[bzoj3991] [SDOI2015]寻宝游戏

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<cstring>
  4 #include<algorithm>
  5 #define ll long long
  6 using namespace std;
  7 const int maxn=100233;
  8 struct zs{int too,pre,dis;}e[maxn<<1];int tot,last[maxn];
  9 int lc[maxn<<1],rc[maxn<<1],p[maxn<<1],rnd[maxn<<1],tt,V,rt,PRE,AFT,st,ed;
 10 int dep[maxn],bel[maxn],fa[maxn],sz[maxn],dfn[maxn],tim;
 11 ll dis[maxn],ans;
 12 int i,j,k,n,m,cnt;
 13 bool is[maxn];
 14  
 15 int ra,fh;char rx;
 16 inline int read(){
 17     rx=getchar(),ra=0,fh=1;
 18     while((rx<‘0‘||rx>‘9‘)&&rx!=‘-‘)rx=getchar();
 19     if(rx==‘-‘)fh=-1,rx=getchar();
 20     while(rx>=‘0‘&&rx<=‘9‘)ra*=10,ra+=rx-48,rx=getchar();return ra*fh;
 21 }
 22  
 23 inline void lturn(int &x){int R=rc[x];rc[x]=lc[R],lc[R]=x,x=R;}
 24 inline void rturn(int &x){int L=lc[x];lc[x]=rc[L],rc[L]=x,x=L;}
 25 void insert(int &x){
 26     if(!x){x=++tt,p[x]=V,rnd[x]=rand()+tt;return;}
 27     if(dfn[V]<dfn[p[x]]){
 28         insert(lc[x]);
 29         if(rnd[lc[x]]<rnd[x])rturn(x);
 30     }else{
 31         insert(rc[x]);
 32         if(rnd[rc[x]]<rnd[x])lturn(x);
 33     }
 34 }
 35 void del(int &x){
 36     if(p[x]==V){
 37         if(!lc[x]||!rc[x])x=lc[x]|rc[x];
 38         else if(rnd[lc[x]]<rnd[rc[x]])rturn(x),del(x);
 39         else lturn(x),del(x);
 40     }else
 41     if(dfn[V]<dfn[p[x]])del(lc[x]);else del(rc[x]);
 42 }
 43 void getpre(int x){
 44     if(!x)return;
 45     if(dfn[p[x]]<dfn[V])PRE=p[x],getpre(rc[x]);
 46     else getpre(lc[x]);
 47 }
 48 void getaft(int x){
 49     if(!x)return;
 50     if(dfn[p[x]]>dfn[V])AFT=p[x],getaft(lc[x]);
 51     else getaft(rc[x]);
 52 }
 53 inline void getst(){
 54     int x=rt;
 55     while(lc[x])x=lc[x];
 56     st=p[x];
 57 }
 58 inline void geted(){
 59     int x=rt;
 60     while(rc[x])x=rc[x];
 61     ed=p[x];
 62 }
 63  
 64  
 65 void dfs(int x){
 66     sz[x]=1,dep[x]=dep[fa[x]]+1;
 67     for(int i=last[x];i;i=e[i].pre)if(e[i].too!=fa[x])
 68         fa[e[i].too]=x,dis[e[i].too]=dis[x]+e[i].dis,dfs(e[i].too),sz[x]+=sz[e[i].too];
 69 }
 70 void dfs2(int x,int chain){
 71     int mx=0,i;dfn[x]=++tim,bel[x]=chain;
 72     for(i=last[x];i;i=e[i].pre)if(e[i].too!=fa[x]&&sz[e[i].too]>sz[mx])mx=e[i].too;
 73     if(!mx)return;
 74     dfs2(mx,chain);
 75     for(i=last[x];i;i=e[i].pre)if(e[i].too!=fa[x]&&e[i].too!=mx)dfs2(e[i].too,e[i].too);
 76 }
 77 inline int getlca(int a,int b){
 78     while(bel[a]!=bel[b]){
 79         if(dep[bel[a]]<dep[bel[b]])swap(a,b);
 80         a=fa[bel[a]];
 81     }
 82     return dep[a]<dep[b]?a:b;
 83 }
 84 inline ll getdis(int a,int b){
 85     int c=getlca(a,b);
 86     return dis[b]-dis[c]+dis[a]-dis[c];
 87 }
 88 inline void insert(int a,int b,int c){
 89     e[++tot].too=b,e[tot].dis=c,e[tot].pre=last[a],last[a]=tot,
 90     e[++tot].too=a,e[tot].dis=c,e[tot].pre=last[b],last[b]=tot;
 91 }
 92 int main(){
 93     n=read(),m=read();
 94     for(i=1;i<n;i++)j=read(),k=read(),insert(j,k,read());
 95     dfs(1),dfs2(1,1);
 96     while(m--){
 97         i=read();
 98         if(!is[i]){
 99             V=i,PRE=AFT=0,getpre(rt),getaft(rt);
100             if(PRE||AFT){
101                 if(!PRE)PRE=ed;if(!AFT)AFT=st;
102                 ans=ans-getdis(PRE,AFT)+getdis(PRE,i)+getdis(i,AFT);
103             }
104             insert(rt),is[i]=1,cnt++;
105         }else{
106             V=i,PRE=AFT=0,getpre(rt),getaft(rt);
107             if(PRE||AFT){
108                 if(!PRE)PRE=ed;if(!AFT)AFT=st;
109                 ans=ans+getdis(PRE,AFT)-getdis(PRE,i)-getdis(i,AFT);
110             }
111             del(rt),is[i]=0,cnt--;
112         }
113         if(cnt)getst(),geted();
114         printf("%lld\n",ans);
115     }
116     return 0;
117 }