校内测 11.4 T2 小王子【LCA】【树上差分】

---

题目：

传送门

---

题意：

有树边和非树边，各选择一条边删除，在保证剩下的图不是一张连通图，有多少种选择方案

---

分析：

这么长的题面真是马都没了
我们假设砍一条树边，这样在不考虑非树边的时候原图就会变成两个连通块，再加入非树边，考虑能满足条件的情况有哪些：
$1.$如果没有边连接这两个连通块，那么我们可以任意选择一条非树边删除，完全不会影响
$2.$如果有且仅有一条边飞出来了，我们就只能选择这一条非树边删除，删除其他的非树边都无法满足不连通的条件
有了这样的性质，再考虑怎么计算答案
对于一条非树边$(x,y)$，会影响到$x\to LCA、LCA\to y$路径上的树边，那么我们可以用树上差分做下，不过要从下到上，不然在子节点$-1$可能会被卡

---

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>  
#include<string>
#include<algorithm>
#include<queue>
#include<ctime>
#include<vector>
#define LL long long
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
struct node{
	LL to,next;
}e[400005];
LL ls[200005],cnt=0;
void add(LL x,LL y)
{
	e[cnt]=(node){y,ls[x]};
	ls[x]=cnt++;
	return;
}
LL fa[200005][20],d[200005],lim;
void pdfs(LL u)
{
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(d[v]) continue;
		d[v]=d[u]+1;
		fa[v][0]=u;
		pdfs(v);
	}
	return;
}
LL LCA(LL x,LL y)
{
	if(d[x]>d[y]) swap(x,y);
	for(LL i=lim;i>=0;i--) if(d[fa[y][i]]>=d[x]) y=fa[y][i];
	if(x==y) return x;
	for(LL i=lim;i>=0;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
LL s[200005],ans=0,n=read(),m=read();
void dfs(LL u,LL fa)
{
	LL tf=0;
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(v==fa) continue;
		dfs(v,u);
		s[u]+=s[v];
	}
	if(fa)
	{
	  if(!s[u]) ans+=m; 
	  else if(s[u]==1) ans++;
	}
	return;
}
int main()
{
	memset(ls,-1,sizeof(ls));
	for(LL i=1;i<n;i++) 
	{
		LL x=read(),y=read();
		add(x,y);add(y,x);
	}
	d[1]=1;
	pdfs(1);
	lim=(int)(log(n)/log(2))+1;
	for(LL j=1;j<=lim;j++)
	  for(LL i=1;i<=n;i++)
	    fa[i][j]=fa[fa[i][j-1]][j-1];
	for(LL i=1;i<=m;i++)
	{
		LL x=read(),y=read();
		s[x]++;s[y]++;
		s[LCA(x,y)]-=2;
	}
	dfs(1,0);
	cout<<ans;
	return 0;
}