【lca】poj3417 Network

我们可以通过枚举每条特殊边对树上边的影响，利用树上差分

对于每个特殊边x->y 给sum[x]+1 sum[y]+1 sum[lca(x,y)]-2 这样就能统计了

 

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+5;
int head[maxn],n,m,cnt,num[maxn],f[maxn][20],dep[maxn];
struct edge
{
	int to,nxt;
}e[maxn<<2];
void add(int x,int y)
{
	e[++cnt].nxt=head[x];
	e[cnt].to=y;
	head[x]=cnt;
}
void dfs(int x,int fa)
{
	dep[x]=dep[fa]+1;
	for(int i=0;i<=18;i++) f[x][i+1]=f[f[x][i]][i];
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==fa) continue;
		f[to][0]=x;
		dfs(to,x);
	}
}
int lca(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=18;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
		if(x==y) return x;
	}
	for(int i=18;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[y][0];
}
int val[maxn];
void calc(int x,int fa)
{
	val[x]+=num[x];
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==fa) continue;
		calc(to,x);
		val[x]+=val[to];
	}
}
int main()
{
	freopen("dark.in","r",stdin);
	freopen("dark.out","w",stdout);
	scanf("%d%d",&n,&m);
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y); add(y,x);
	}
	dfs(1,0);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		num[x]++; num[y]++;
		num[lca(x,y)]-=2;
	}
	calc(1,0);
	int ans=0;
	for(int i=2;i<=n;i++)
	{
		if(!val[i]) ans+=m;
		if(val[i]==1) ans++;
	}
	printf("%d\n",ans);
	return 0;
}