[JZOJ100019]A--dfn序+扫描线

[JZOJ100019]A--dfn序+扫描线

题目链接

太懒了自行搜索

分析

这道题查了一个下午的错,真的心态崩了

不过这道题确实妙啊

类比于喝喝喝,我们发现任何一条覆盖了非法路径的路径一定不合法,假如非法路径为\((x,kx)\),设\(u\)为\(dfn\)序中较小的那个点,\(v\)为较大的那个;假如覆盖了路径\((u,v)\)的路径为\((a,b)(dfn_a<dfn_b)\);

设数组\(ed[x]\)为\(x\)的子树中\(dfs\)最大的那个点的\(dfs\)序,显然这可以和\(dfs\)序在一遍\(dfs\)中求出

我们还需要知道:\(x\) 的子树中\(dfs\)序是连续的一段

考虑两种情况:

• \(u\)是\(lca(u,v)\),分析发现这也有两种情况
  
  先设\(g\)为路径\((u,v)\)上距\(u\)最近的那个点

  • Case#1
    
    \(dfn[v]<=dfn[a]<=ed[v]\) , \(dfn[b]>ed[g]\)
    
    

  • Case#2

  \(dfn[v]<=dfn[b]<=ed[v]\) , \(dfn[g]>dfn[a]\)
  
  

• \(u\)不是\(lca(u,v)\)

  这个只有一种情况,\(a,b\)分别在\(u,v\)子树中,即

  Case#3

  \(dfn[u]<=dfn[a]<=ed[u]<=dfn[v]<=dfn[b]<=ed[v]\)

我们把\((a,b)\)看成一个有序数对的话,发现其实每一种\(case\)都围成了一个矩形,我们只需要求出矩形面积的并之后即可得出所有不合法的路径,用总路径数\(n*(n-1)/2\)减去不合法即所求

求矩形面积的并使用线段树的扫描线法,但是发现一个点我们也要算进贡献,并非严格意义的二维图形；

因为值域很小\((1e5)\),学会了一种船新操作,我们不排序,将纵坐标相同的丢入一个容器,直接从小到大钦定纵坐标.不断取出对应容器内的线段加入,然后计算剩余的点数即可(纵坐标已钦定)

一道图论题转化成了数据结构,也是精妙啊

代码

/*
  code by RyeCatcher
*/
inline char gc(){
    static char buf[SIZE],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,SIZE,stdin),p1==p2)?EOF:*p1++;
}
template <class T>inline void read(T &x){
    x=0;int ne=0;char c;
    while((c=gc())>'9'||c<'0')ne=c=='-';x=c-48;
    while((c=gc())>='0'&&c<='9')x=(x<<3)+(x<<1)+c-48;x=ne?-x:x;return ;
}
const int maxn=500005;
const int inf=0x7fffffff;
int n;
struct Edge{
	int ne,to;
}edge[maxn<<1];
int h[maxn],num_edge=1;
int fa[maxn][17];
inline void add_edge(int f,int to){
	edge[++num_edge].ne=h[f];
	edge[num_edge].to=to;
	h[f]=num_edge;
}
int dfn[maxn],ed[maxn],dep[maxn],tot=0;
void dfs(int now){
	int v;dfn[now]=++tot;
	for(ri i=h[now];i;i=edge[i].ne){
		v=edge[i].to;
		if(v==fa[now][0]||dfn[v])continue;
		dep[v]=dep[now]+1,fa[v][0]=now;
		for(ri i=1;i<=16;i++)fa[v][i]=fa[fa[v][i-1]][i-1];
		dfs(v);
	}
	ed[now]=tot;
	return ;
}
int get_g(int x,int y){//x到y路径上距y最近
	for(ri i=16;i>=0;i--){
		//printf("%d %d\n",fa[x][i],x);
		if(dep[fa[x][i]]>dep[y]){
			x=fa[x][i];
			//printf("%d\n",x);
		}
	}
	return x;
}
int L,R,dta;
struct Seg{
	int l,r,h,d;
	Seg(){l=r=h=d=0;}
	Seg(int _l,int _r,int _h,int _d){l=_l,r=_r,h=_h,d=_d;}
}seg[maxn<<4];
int poi=0;
ll sum[maxn<<2];
ll tag[maxn<<2];
vector <int> dd[maxn];
inline void modify(int now,int l,int r){
	if(tag[now]>0)sum[now]=(r-l+1);
	else sum[now]=sum[now<<1]+sum[now<<1|1];
}
void update(int now,int l,int r){
	if(L<=l&&r<=R){
		tag[now]+=dta;
		modify(now,l,r);
		return ;
	}
	int mid=(l+r)>>1;
	if(L<=mid)update(now<<1,l,mid);
	if(mid<R)update(now<<1|1,mid+1,r);
	modify(now,l,r);
	return ;
}
ll ans=0;
int main(){
	int x,y,ex,ey,g;
	FO(a);
	//freopen("a5.in","r",stdin);
	read(n);
	for(ri i=1;i<n;i++){
		read(x),read(y);
		add_edge(x,y);
		add_edge(y,x);
	}
	dep[1]=1,fa[1][0]=0;
	dfs(1);
	int p,q;
	for(ri i=1;i<=n;i++){
		for(ri j=i+i;j<=n;j+=i){
			p=i,q=j;
			if(dfn[p]<dfn[q])swap(p,q);
			x=dfn[p],y=dfn[q];
			ex=ed[p],ey=ed[q];
			if(x>=y&&x<=ey){
				g=get_g(p,q);
				dd[x].push_back(++poi);
				seg[poi]=Seg(1,dfn[g]-1,x,1);
				dd[ex+1].push_back(++poi);
				seg[poi]=Seg(1,dfn[g]-1,ex,-1);
				if(ed[g]==n)continue;
				dd[ed[g]+1].push_back(++poi);
				seg[poi]=Seg(x,ex,ed[g]+1,1);
				dd[n+1].push_back(++poi);
				seg[poi]=Seg(x,ex,n,-1);
			}
			else {
				dd[x].push_back(++poi);
				seg[poi]=Seg(y,ey,x,1);
				dd[ex+1].push_back(++poi);
				seg[poi]=Seg(y,ey,ex,-1);
			}
		}
	}
	for(ri i=1;i<=n;i++){
		for(ri j=0;j<dd[i].size();j++){
			x=dd[i][j];
			L=seg[x].l,R=seg[x].r,dta=seg[x].d;
			update(1,1,n);
		}
		ans+=sum[1];
	}
	printf("%lld\n",1ll*n*(n-1)/2-ans);
	return 0;
}