JZOJ 6276.树【扫描线】【线段树】

---

题目：

传送门

---

题意：

有一棵无根树，现在有$m$个点对$x,y$，要求在任意$u,v$的简单路径上不能既经过$x$又经过$y$
问一共有多少路径满足要求

---

分析：

正着推看起来思维难度很高，所以我们喜闻乐见的反着推，找出不合法的方案数就好了
我们先求出无根树的$dfs$序，在输入每个点对后，我们结合两个点的$dfs$序，用扫描线形成一个矩形，在矩形内的所有点都是不合法的
再然后我们用线段树来维护我们的扫描线

---

代码：

#pragma GCC optimize("Ofast")
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cctype>
#include<cmath>
#define LL long long 
#define LZX Mu
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;
}
int n,m,fir[100005];
int cnt,tot,ls[200005],v[100005],sec[100005];
int num=0,f[100005][30],dep[100005],T;
LL ans;
struct tree{
    int l,r,mark,w;
}t[800005];
struct line_node{
	int x,Y1,Y2,w;
}line[400005];
struct edge{
	int to,next;
}a[200005];
void build(int k,int l,int r)
{
    t[k].l=l;t[k].r=r;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
    return;
}
void change(int k,int l,int r,int num)
{
    if(t[k].l==l&&t[k].r==r)
    {
        t[k].mark+=num;
        if(t[k].mark) t[k].w=t[k].r+1-t[k].l;
        else if(l==r) t[k].w=0;
        else t[k].w=t[k*2].w+t[k*2+1].w;
        return;
    }
    int mid=(t[k].l+t[k].r)>>1;
    if(r<=mid) change(k*2,l,r,num);
    else if(l>mid) change(k*2+1,l,r,num);
    else change(k*2,l,mid,num),change(k*2+1,mid+1,r,num);
    if(t[k].mark) t[k].w=t[k].r+1-t[k].l;
    else t[k].w=t[k*2].w+t[k*2+1].w;
    return;
}
void add1(int x,int y)
{
	a[tot]=(edge){y,ls[x]};
	ls[x]=tot++;
	return;
}
void add2(int x1,int x2,int Y1,int Y2)
{
	if(x1>x2) swap(x1,x2);
	if(Y1>Y2) swap(Y1,Y2);
	line[++num]=(line_node){x1,Y1,Y2,1};
	line[++num]=(line_node){x2+1,Y1,Y2,-1};
	return;
}
void dfs(int x,int fa)
{
	cnt++;
	fir[x]=cnt;
	f[x][0]=fa;
	for(register int i=ls[x];~i;i=a[i].next)
	{
		int y=a[i].to;
		if(y==fa) continue;
		dep[y]=dep[x]+1;
		dfs(y,x);
	}
	sec[x]=cnt;
	return;
}
int MARMOT(int x,int y)
{
    for(register int i=T;i>=0;i--)
      if(dep[f[y][i]]>dep[x]) y=f[y][i];
    return y;
}
bool cmp(line_node x,line_node y) {return x.x<y.x;}
int main()
{
/*	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);*/
	memset(ls,-1,sizeof(ls));
	n=read();m=read();
	for(register int i=1;i<n;i++)
	{
		int x=read(),y=read();
		add1(x,y);add1(y,x);
	}
	dfs(1,0);
	T=(int)(log(n)/log(2))+1;
    for(register int j=1;j<=T;j++)
      for(register int i=1;i<=n;i++)
        f[i][j]=f[f[i][j-1]][j-1];
	for(register int i=1;i<=m;i++)
	{
		int x=read(),y=read();
		if(fir[x]>fir[y]) swap(x,y);
		if(fir[y]<=sec[x]&&fir[y]>fir[x])
		{
			int son=MARMOT(x,y);
			if(fir[son]!=1) add2(1,fir[son]-1,fir[y],sec[y]);
			if(sec[son]!=n) add2(fir[y],sec[y],sec[son]+1,n);
		}
		else add2(fir[x],sec[x],fir[y],sec[y]);
	}
	sort(line+1,line+1+num,cmp);
	build(1,1,n);
	int l=1;
	for(register int i=1;i<=n;i++)
	{
		while(l<=num&&line[l].x<=i)
		{
			change(1,line[l].Y1,line[l].Y2,line[l].w);
			l++;
		}
		ans+=t[1].w;
	}
	cout<<(LL)n*(n-1)/2-ans;
	return 0;
}