#扫描线+线段树# [ssloj 1459] 空间简单度

Title

---

Solution

[jzoj 6276] 【noip提高组模拟1】树 {扫描线+线段树}

---

Code

#include<cstdio>
#include<stdlib.h>
#include<cmath>
#include<algorithm>
#define rep(i,x,y) for(register int i=x;i<=y;i++)
using namespace std; 
const int N=300100; 
struct node{int y,next;}a[N*2];
struct nodd{int l,r,wo,lazy;}b[N*4];
struct noda{int x,y1,y2,w;}c[N*40]; 
int m,n,fa[N][30],beg[N],ed[N],num,T,tot,head[N],cnt,dep[N]={0,1},k=1; 
long long ans; 

void build(int l,int r,int kg){
   b[kg].l=l; b[kg].r=r; 
   if (l==r) return; 
   int mid=(l+r)/2; 
   build(l,mid,kg*2); build(mid+1,r,kg*2+1); 
}
void update(int l,int r,int kg,int numm){
    if (b[kg].l==l&&b[kg].r==r){
    	b[kg].lazy+=numm; 
   		if (b[kg].lazy) b[kg].wo=b[kg].r-b[kg].l+1; 
    	else if (l==r) b[kg].wo=0; 
    	else b[kg].wo=b[kg*2].wo+b[kg*2+1].wo; 
    	return; 
	}
    if (b[kg*2].r>=r) update(l,r,kg*2,numm); 
    else if (b[kg*2+1].l<=l) update(l,r,kg*2+1,numm); 
    else { update(l,b[kg*2].r,kg*2,numm); update(b[kg*2+1].l,r,kg*2+1,numm); }
    if (b[kg].lazy) b[kg].wo=b[kg].r-b[kg].l+1; 
	else b[kg].wo=b[kg*2].wo+b[kg*2+1].wo; 
}

void dfs(int x,int f){
	fa[x][0]=f; beg[x]=++cnt; 
	for(register int i=head[x];i;i=a[i].next){
		int y=a[i].y; 
		if (y!=f){
			dep[y]=dep[x]+1; 
			dfs(y,x); 
		}
	}
	ed[x]=cnt; 
}
int Lca(int x,int y){
	for(int i=T;i>=0;i--)
	if (dep[fa[y][i]]>dep[x]) y=fa[y][i]; 
	return y; 
}

void add(int x1,int x2,int y1,int y2){
	if (x1>x2) swap(x1,x2); 
	if (y1>y2) swap(y1,y2); 
	c[++num]=(noda){x1,y1,y2,1}; 
	c[++num]=(noda){x2+1,y1,y2,-1}; 
}

bool operator<(noda x,noda y){return x.x<y.x;}
int main(){
	int size = 256 << 20; 
	char*p=(char*)malloc(size) + size;
	__asm__("movl %0, %%esp\n" :: "r"(p) );
	 
	scanf("%d%d",&n,&m); 
	T=(int)(log(n)/log(2))+1; 
	rep(i,1,n-1){
		int x,y; 
		scanf("%d%d",&x,&y); 
		a[++tot]=(node){y,head[x]}; head[x]=tot; 
		a[++tot]=(node){x,head[y]}; head[y]=tot; 
	}
	dfs(1,0); 
	rep(j,1,T) rep(i,1,n) fa[i][j]=fa[fa[i][j-1]][j-1]; 

	rep(i,1,n) rep(j,i+1,min(i+m,n)){		
		int x=i,y=j; 
		if(beg[x]>beg[y]) swap(x,y);
		if(beg[y]<=ed[x]&&beg[y]>beg[x]){
			int s=Lca(x,y);
			if(beg[s]!=1) add(1,beg[s]-1,beg[y],ed[y]);
			if(ed[s]!=n) add(beg[y],ed[y],ed[s]+1,n);
		}
		else add(beg[x],ed[x],beg[y],ed[y]);
	}
	sort(c+1,c+num+1); 
	build(1,n,1);
	rep(i,1,n){
		while (k<=num&&c[k].x<=i){
			update(c[k].y1,c[k].y2,1,c[k].w);
			k++; 
		}
		ans+=b[1].wo; 
	}
	printf("%lld",1ll*n*(n-1)/2-ans+n); 
}