其实就是求某一个节点到生成树的某条链的最短距离,用dfs序判断这条链的两个端点就好了。 #include #include #include #include #include #include using namespace std; typedef pair pd; typedef map ::iterator it; #define clr(a) memset(a,0,sizeof(a)) struct edg{int v,w,nx;}gp[200010]; int hd[100010],cnt,n,q,dfsi,dfsr[100010]; int fs[100010][20],dep[100010],ds[100010]; void psh(int u,int v,int w){ ++cnt; gp[cnt].v=v,gp[cnt].w=w; gp[cnt].nx=hd[u],hd[u]=cnt; }; map mp; void dfs_(int v,int d,int sd){ int i;dep[v]=d,ds[v]=sd; dfsr[v]=dfsi++; for(i=hd[v];i;i=gp[i].nx){ if(gp[i].v!=fs[v][0]){ fs[gp[i].v][0]=v; dfs_(gp[i].v,d+1,sd+gp[i].w); } } }; void genfs(){ int i,j,k; for(k=2,j=1;k <<=1,++j) for(i=1;i<=n;++i) fs[i][j]=fs[fs[i][j-1]][j-1]; }; int trs(int v,int d){ for(int i=0;d;d>>=1,++i) if(d&1) v=fs[v][i]; return v; }; int lca(int x,int y){ if(dep[x]!=dep[y]){ if(dep[x]>dep[y]) x=trs(x,dep[x]-dep[y]); else y=trs(y,dep[y]-dep[x]); } if(x==y)return x; for(int i=17;i>=0;--i) if(fs[x][i]!=fs[y][i]) x=fs[x][i],y=fs[y][i]; return fs[x][0]; }; int cl(int v){ if(mp.empty())return 0; it t=mp.upper_bound(dfsr[v]); int x,y; if(t!=mp.end()&&t!=mp.begin()){ x=t->second;--t; y=t->second; } else{ x=mp.begin()->second; y=mp.rbegin()->second; } return ds[v]-ds[lca(x,v)]-ds[lca(y,v)]+ds[lca(x,y)]; }; bool ck(int a,int b){ if(a==1){ if(mp.find(dfsr[b])!=mp.end()) return 0; } else{ if(mp.find(dfsr[b])==mp.end()) return 0; } return 1; }; int ct=0; void solve(){ int i,j,a,b,c; scanf("%d %d",&n,&q); clr(hd),cnt=0; for(i=1;i