Jzoj3717【NOI2014模拟7.2】火车

A国有n个城市，城市之间有一些双向道路相连，并且城市两两之间有唯一路径。现在有火车在城市a，需要经过m个城市。火车按照以下规则行驶：每次行驶到还没有经过的城市中在m个城市中最靠前的。现在小A想知道火车经过这m个城市后所经过的道路数量。

很显然的树剖题目嘛，加上个区间打标记即可，可以用树状数组

注意数据很坑，dfs爆栈，所以要从最中间开始dfs

（真的不知道为什么是NOI模拟）

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 500010
using namespace std;
struct Edge{ int v,nt; } G[N<<1];
int h[N],f[N],d[N],top[N],son[N],sz[N];
int n,m,cnt=0,k,w[N],v[N],tot=0; long long ans=0;
inline void adj(int x,int y){
	G[++cnt]=(Edge){y,h[x]}; h[x]=cnt;
}
inline int _(int x){ return x&-x; }
inline void add(int x,int k){ for(;x<=n;x+=_(x)) v[x]+=k; }
inline int sum(int x,int k=0){ for(;x;x-=_(x)) k+=v[x]; return k; }
void dfs(int x,int p){
	f[x]=p; d[x]=d[p]+1; sz[x]=1;
	for(int v,i=h[x];i;i=G[i].nt)
		if((v=G[i].v)!=p){
			dfs(v,x);
			sz[x]+=sz[v]; 
			if(sz[v]>sz[son[x]]) son[x]=v;
		}
}
void dijk(int x,int t){
	top[x]=t; w[x]=++tot;
	if(son[x]) dijk(son[x],t);
	for(int v,i=h[x];i;i=G[i].nt)
		if((v=G[i].v)!=f[x]&&v!=son[x]) dijk(v,v);
}
int Lca(int a,int b){
	for(;;){
		if(top[a]==top[b]){
			if(d[a]>d[b]) swap(a,b);
			add(w[b]+1,-1); add(w[a],1);
			return a;
		}
		if(d[top[a]]>d[top[b]]) swap(a,b);
		add(w[b]+1,-1); add(w[top[b]],1); b=f[top[b]];
	}
}
int range(int a,int b){
	if(sum(w[b])) return -1;
	int c=Lca(a,b);
	return d[a]+d[b]-2*d[c];
}
int main(){
	freopen("train.in","r",stdin);
	freopen("train.out","w",stdout);
	scanf("%d%d%d",&n,&m,&k);
	for(int a,b,i=1;i<n;++i) { scanf("%d%d",&a,&b); adj(a,b); adj(b,a); }
	dfs(n>>1,0); dijk(n>>1,n>>1);
	for(int x,i=0;i<m;++i){
		scanf("%d",&x);
		int b=range(k,x);
		if(b>0){ k=x;ans+=b; }
	}
	printf("%lld\n",ans);
}