#10131. 「一本通 4.4 例 2」暗的连锁

传送门

树上差分与树上倍增

注意是边差分,所以压边权入点权

边差分$:w[x]++,w[y]++,w[lca(w,y)]-=2$

在边差分中，$w[x]表示x号点向上的边的权值$

点差分$:w[x]++,w[y]++,w[lca(w,y)]-=1,w[fa[lac(x,y)]]-=1$

在点差分中，$w[x]表示x节点的权值$

Code

#include<bits/stdc++.h>
#define rep(i,a,b) for(register int (i)=(a);(i)<=(b);(i)++) 
#define don(i,a,b) for(register int (i)=(a);(i)>=(b);(i)--)
using namespace std;
const int maxn=2e5+10;
const int maxm=1e3+10;
int ans=0,n,m,cnt;
int f[maxn][21],dep[maxn],head[maxn],w[maxn],a[maxn],b[maxn],l[maxn];

template <class t> inline void read(t &x)
{
    x=0;int f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=10*x+ch-'0';ch=getchar();}
    x*=f;
}

struct node{
	int v,nex;
}e[maxn<<1];

void add(int u,int v) {
	e[++cnt].v=v;
	e[cnt].nex=head[u];
	head[u]=cnt;
}

void readdata() {
	read(n);read(m);
	rep(i,1,n-1) {
		int x,y;
		read(x),read(y);
		add(x,y);
		add(y,x);
	}
	rep(i,1,m) {
		read(a[i]),read(b[i]);
		w[a[i]]++;w[b[i]]++;
	}
}

void pre(int u,int fa) {
	dep[u]=dep[fa]+1;
	rep(i,0,19) f[u][i+1]=f[f[u][i]][i];
	for(int i=head[u];i;i=e[i].nex) {
		int v=e[i].v;
		if(v==fa) continue;
		f[v][0]=u;
		pre(v,u);
	}
}

inline int lca(int x,int y) {
	if(dep[x]<dep[y]) swap(x,y);
	don(i,20,0) {
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
		if(x==y) return x;
	}
	don(i,20,0) {
		if(f[x][i]!=f[y][i]) {
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}

void dfs(int u,int fa) {
	for(int i=head[u];i;i=e[i].nex) {
		int v=e[i].v;
		if(v==fa) continue;
		dfs(v,u);
		w[u]+=w[v];
	}
}

void work() {
//	memset(w,false,sizeof(w));
	memset(dep,false,sizeof(dep));
	pre(1,0);
	rep(i,1,m) {
		int ancestor=lca(a[i],b[i]);
		l[i]=lca(a[i],b[i]);
		w[ancestor]-=2;
		//printf("%d\n",l[i]);
	}
	dfs(1,0);
	rep(i,2,n) {
		if(w[i]==0) ans+=m;
		if(w[i]==1) ans++;
	}
	printf("%d\n",ans);
}

int main() {
	freopen("input.txt","r",stdin);
//	freopen("a.txt","w",stdout);
	readdata();
	work();
	return 0;
}