[CTSC2018]暴力写挂

题目

边分治+虚树=双倍的快乐

这个柿子里有两个(LCA),我们考虑魔改一下前面的(operatorname{depth(x)+depth(y)-depth(LCA(x,y))}),为了方便边分,我们考虑把(operatorname{depth(LCA(x,y))})去掉变换为树上距离

经过一番魔改,这个柿子变成了(frac{1}{2}(operatorname{depth(x)+depth(y)+dis(x,y)}))

至于第二棵树上的(operatorname{LCA}),我们只能考虑到虚树上去搞了

对于当前的分治边(w),我们处理处分块内所有点(i)到这条边的距离(p_i),那么(operatorname{depth(x)+depth(y)+dis(x,y)}=p_x+p_y+w+depth_x+depth_y)

同时我们把分治边左右两边的点黑白染色,点权设为(p_i+depth_i),到第二颗树上去建虚树,对于虚树上的每个点考虑其作为(operatorname{LCA})时的贡献,于是把问题转化成了在一个点的不同儿子里找到一对异色点,使得点权最大,显然直接虚树上dp就好了

如果虚树实现得不好,这个算法就是(operatorname{O(nlog^2n)})

如果使用st表求(operatorname{LCA}),并且在一开始就把所有点按照(operatorname{dfn})排序,每次根据这个点染成的颜色分到两个接下来的分治块里去,就能做到线性建虚树,整个算法的复杂度就能到(operatorname{O(nlogn)})

记得特判(x=y)的情况

代码

#include<bits/stdc++.h>
#define re register
#define LL long long
#define pt putchar(1)
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=366667;
const int M=maxn*4;
const LL inf=-1e15;
inline int read() {
	char c=getchar();int x=0,r=1;
	while(c<'0'||c>'9') {if(c=='-') r=-1;c=getchar();}
	while(c>='0'&&c<='9') x=(x<<3ll)+(x<<1ll)+c-48,c=getchar();return r*x;
}
std::vector<int> son[M];
std::vector<LL> v[M];
struct E{int v,nxt,w;}e[M<<1];
int head[M],sum[M],vis[M],dfn[maxn],b[maxn];
int S,mnow,rn,n,num,col[maxn],rt,Mnow,tp[2],c[2][maxn];
LL g[maxn],dp[maxn][2],deep[maxn],tmp,ans;
inline int cmp(int A,int B) {return dfn[A]<dfn[B];}
inline void add(int x,int y,LL w) {
	e[++num].v=y;e[num].nxt=head[x];head[x]=num,e[num].w=w;
}
void dfs1(int x,int fa) {
	for(re int i=head[x];i;i=e[i].nxt) {
		if(e[i].v==fa) continue;
		deep[e[i].v]=deep[x]+e[i].w;
		son[x].push_back(e[i].v);v[x].push_back(e[i].w);
		dfs1(e[i].v,x);
	}
} 
void getrt(int x,int fa) {
	sum[x]=1;
	for(re int i=head[x];i;i=e[i].nxt) {
		if(vis[i>>1]||e[i].v==fa) continue;
		getrt(e[i].v,x);sum[x]+=sum[e[i].v];
		int now=max(sum[e[i].v],S-sum[e[i].v]);
		if(now<Mnow) Mnow=now,rt=i;
	}
}
struct Virtual_Tree {
	struct E{int v,nxt,w;}e[maxn<<1];
	int pos[maxn],f[21][maxn<<1],lg[maxn<<1];
	LL pre[maxn];int num,cnt,tot,head[maxn],top,st[maxn],d[maxn],root;
	inline void add(int x,int y,LL w) {
		e[++num].v=y;e[num].nxt=head[x];head[x]=num;e[num].w=w;
	}
	void dfs(int x) {
		dfn[x]=++cnt,f[0][++tot]=x;pos[x]=tot;
		for(re int i=head[x];i;i=e[i].nxt) {
			if(dfn[e[i].v]) continue;
			pre[e[i].v]=pre[x]+e[i].w;d[e[i].v]=d[x]+1;
			dfs(e[i].v);f[0][++tot]=x;
		}
	}
	inline int LCA(int x,int y) {
		int l=pos[x],r=pos[y];
		if(l>r) std::swap(l,r);
		int k=lg[r-l+1];
		if(d[f[k][l]]<d[f[k][r-(1<<k)+1]]) return f[k][l];
		return f[k][r-(1<<k)+1];
	}
	inline void build() {
		for(re int x,y,w,i=1;i<rn;i++) {
			x=read(),y=read(),w=read();
			add(x,y,w),add(y,x,w);
		}
		d[1]=1;dfs(1);memset(head,0,sizeof(head));
		for(re int i=2;i<=tot;i++) lg[i]=lg[i>>1]+1;
		for(re int j=1;j<=lg[tot];j++)
			for(re int i=1;i+(1<<(j-1))<=tot;i++)
				if(d[f[j-1][i]]<d[f[j-1][i+(1<<(j-1))]]) f[j][i]=f[j-1][i];
					else f[j][i]=f[j-1][i+(1<<(j-1))];
		for(re int i=1;i<=rn;i++) b[i]=i;
		std::sort(b+1,b+rn+1,cmp);
	}
	inline void ins(int x) {
		if(top<1) {st[++top]=x;return;}
		int lca=LCA(x,st[top]);
		if(lca==st[top]) {st[++top]=x;return;}
		while(top>1&&dfn[st[top-1]]>=dfn[lca]) 
			add(st[top-1],st[top],0),top--;
		if(lca!=st[top]) add(lca,st[top],0),st[top]=lca;
		st[++top]=x;
	}
	void tree_dp(int x) {
		dp[x][col[x]]=g[x];
		for(re int i=head[x];i;i=e[i].nxt) {
			tree_dp(e[i].v);
			tmp=max(tmp,-2ll*pre[x]+dp[x][0]+dp[e[i].v][1]);
			tmp=max(tmp,-2ll*pre[x]+dp[x][1]+dp[e[i].v][0]);
			dp[x][0]=max(dp[x][0],dp[e[i].v][0]);
			dp[x][1]=max(dp[x][1],dp[e[i].v][1]);
		}
	}
	inline void get_tree(int l,int r) {
		top=0;num=0;
		if(b[l]!=1) ins(1);
		for(re int i=l;i<=r;i++) ins(b[i]);
		while(top>1) add(st[top-1],st[top],0),top--;
		tree_dp(1);
	}
	void del(int x) {
		for(re int i=head[x];i;i=e[i].nxt) del(e[i].v);
		dp[x][0]=dp[x][1]=g[x]=inf;head[x]=0;
	}
}T;
void dfs2(int x,int fa,int o,LL d) {
	if(x<=rn) col[x]=o,g[x]=deep[x]+d;
	for(re int i=head[x];i;i=e[i].nxt) {
		if(vis[i>>1]||e[i].v==fa) continue;
		dfs2(e[i].v,x,o,d+e[i].w);
	}
}
void solve(int x,int s,int l,int r) {
	if(l>r) return;
	Mnow=M,S=s,getrt(x,0);
	if(Mnow==M) return;vis[rt>>1]=1;
	dfs2(e[rt].v,0,0,0),dfs2(e[rt^1].v,0,1,e[rt].w);
	tmp=inf;T.get_tree(l,r);
	ans=max(ans,tmp);
	tp[0]=tp[1]=0;T.del(1);
	for(re int i=l;i<=r;i++) c[col[b[i]]][++tp[col[b[i]]]]=b[i];
	for(re int i=l,k=1;k<=tp[0];i++,k++) b[i]=c[0][k];
	for(re int i=l+tp[0],k=1;k<=tp[1];i++,k++) b[i]=c[1][k];
	int now=s-sum[e[rt].v],k=rt,L=l+tp[0]-1,R=r-tp[1]+1;
	solve(e[k].v,sum[e[k].v],l,L);
	solve(e[k^1].v,now,R,r);
}
int main() {
	rn=n=read();
	for(re int w,x,y,i=1;i<n;i++) {
		x=read(),y=read(),w=read();
		add(x,y,w),add(y,x,w);
	}
	dfs1(1,0);num=1;memset(head,0,sizeof(head));
	int s[2];
	for(re int i=1;i<=n;i++) {
		int t=son[i].size();
		if(!t) continue;
		if(t==1) {
			add(i,son[i][0],v[i][0]),add(son[i][0],i,v[i][0]);
			continue;
		}
		if(t==2) {
			add(i,son[i][0],v[i][0]),add(son[i][0],i,v[i][0]);
			add(son[i][1],i,v[i][1]),add(i,son[i][1],v[i][1]);
			continue;
		}
		s[0]=++n,s[1]=++n;
		add(i,s[0],0),add(s[0],i,0);add(i,s[1],0),add(s[1],i,0);
		for(re int j=0;j<v[i].size();j++)
			son[s[j&1]].push_back(son[i][j]),v[s[j&1]].push_back(v[i][j]);
	}
	T.build();ans=inf;
	for(re int i=1;i<=rn;i++) dp[i][0]=dp[i][1]=g[i]=inf;
	solve(1,n,1,rn);ans>>=1ll;
	for(re int i=1;i<=n;i++) ans=max(ans,deep[i]-T.pre[i]);
	std::cout<<ans;
	return 0;
}
原文地址:https://www.cnblogs.com/asuldb/p/11304888.html