codeforces 191C Fools and Roads (树链剖分 或 树上差分) HQG_AC 的博客

#include <bits/stdc++.h>
using namespace std ;
const int N = 100000 + 10 ;
struct node{
	int v,next,en ;
}e[2*N];
int cf[N] ;
int ans[N] ;
bool vis[N] ;
int tot,a,b,n,u,v,k ;
int head[N],deep[N] ;
int f[N][32] ;
void add_edge(int u,int v,int en)
{
	e[tot].en=en ;
	e[tot].v=v ;
	e[tot].next=head[u] ;
	head[u]=tot++;
}
//两个点的地方打个标记 lca的地方再打个标记 然后最后dfs一遍统计下标记
void dfs(int u,int fa) {
	deep[u]=deep[fa]+1 ;
	f[u][0]=fa ;//表示x向上走了2^0(=1)部到达的点 
	for (int i=1;(1<<i)<=deep[u];i++) f[u][i]=f[f[u][i-1]][i-1] ;
	/*因为在循环到i时f[x][i-1]一定已经求好，
	又因为f[x]的2^i祖先=f[x]的2^(i-1)祖先的2^(i-1)祖先，所以这么求 */
	for (int i=head[u];i!=-1;i=e[i].next) {
		int v=e[i].v ;
		if (v!=fa) dfs(v,u) ;
	}
}
int LCA(int a,int b){ //最近公共祖先 
	if (deep[a]>deep[b]) swap(a,b) ;
	for (int i=31;i>=0;i--) //把a和b放到同一深度 
		if (deep[a]<=deep[b]-(1<<i)) b=f[b][i] ;
	if (a==b) return a;
	for (int i=31;i>=0;i--){
		if (f[a][i]==f[b][i]) continue ;
		else a=f[a][i],b=f[b][i] ;
	} 
	return f[a][0] ;
} 
void make_cf(int u){
	vis[u]=true ;
	for (int i=head[u];i!=-1;i=e[i].next){
		int v=e[i].v ;
		if (vis[v]) continue ; 
		make_cf(v) ;
		cf[u]+=cf[v] ;
		ans[e[i].en]+=cf[v] ;
	}
}
int main()
{
	memset(head,-1,sizeof(head)) ;
	
	scanf("%d",&n) ;
	for (int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v) ;
		add_edge(u,v,i) ;
		add_edge(v,u,i) ;
	}
	dfs(1,0) ;
	scanf("%d",&k) ;
	for (int i=1;i<=k;i++)
	{
		scanf("%d%d",&u,&v) ;
		cf[u]++;cf[v]++;cf[LCA(u,v)]-=2 ; 
	}
	memset(vis,false,sizeof(vis)) ;
	memset(ans,0,sizeof(ans)) ;
	make_cf(1) ;
	for (int i=1;i<n;i++) printf("%d ",ans[i]) ;
}