牛客练习赛67 F.牛妹的苹果树(树的直径/倍增 线段树)

题目

牛妹种了一棵苹果树。

这棵苹果树有n(n<=3e5)个节点，n-1条边，每一条边都有一个权值wi(1<=wi<=1e9)​。

我们定义：这棵树上的两点之间距离dist(u,v)为它们简单路径上所有边的权值和。

现在，牛妹想给你q(q<=3e5)次询问，每次询问一个区间[l,r]，求。

思路来源

syh0313代码

题解

首先，考虑如果有两个连通块，如何合并，

(a1,b1)是第一个连通块直径，(a2,b2)是第二个连通块直径，

则合并之后的连通块直径不会劣于(a1,b1)(a2,b2)(a1,a2)(b1,b2)(a1,b2)(b1,a2)六者的最大值，

反证法可以考虑第一个连通块取一个点c，第二个连通块取一个点d，

考虑a2、b2、d三个点路径的交点p，|a2p|>|dp|，|b2p|>|dp|

则无论p向哪个方向移动时，max(|a2p|,|b2p|)>|dp|，

这表明当p挪到与c重合时，将d换为a2或b2中的一个点更优，

同理可以把c换成(a1,b1)中的点

 

于是就可以合并了，先ST表RMQ预处理lca，

ans[i][j]表示i后长为1<<j的区间的连通块（只考虑这些点号）的直径的点对(a,b)

倍增预处理步长，合并时对六者取最大值做合并，

查询时，即对覆盖的两个最大子区间做合并即可，

也可以用线段树实现，父节点的区间为lson和rson的合并

代码

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
typedef pair<int,int> P;
typedef long long ll;
const int N=3e5+10,M=2*N;
vector<P>E[N];
int n,q,u,v,w,l,r;
int lg2[M],dfn[M],tot,dep[M],id[N],dp[M][22];
P ans[N][22];
ll dis[N];
void dfs(int u,int fa,int d){
	id[u]=++tot;
	dfn[tot]=u;
	dep[tot]=d;
    for(int i=0;i<E[u].size();++i){
		int v=E[u][i].fi,w=E[u][i].se;
		if(v==fa)continue;
		dis[v]=dis[u]+w;
		dfs(v,u,d+1);
		dfn[++tot]=u;
		dep[tot]=d;
	}
}
void ST(){
	for(int i=1;i<=tot;++i){
        dp[i][0]=i;
	}
	for(int len=1;(1<<len)<=tot;++len){
		for(int l=1;l+(1<<len)-1<=tot;++l){
			if(dep[dp[l][len-1]]<dep[dp[l+(1<<(len-1))][len-1]])dp[l][len]=dp[l][len-1];
			else dp[l][len]=dp[l+(1<<(len-1))][len-1];
		}
	}
}
int RMQ(int l,int r){//返回最小值下标
	int len=lg2[r-l+1];
	if(dep[dp[l][len]]<dep[dp[r-(1<<len)+1][len]])return dp[l][len];
	else return dp[r-(1<<len)+1][len];
}
int lca(int u,int v){
	int mn=min(id[u],id[v]);
	int mx=max(id[u],id[v]);
	return dfn[RMQ(mn,mx)];
}
ll cal(int u,int v){
    return dis[u]+dis[v]-2ll*dis[lca(u,v)];
}
bool upd(ll &x,ll y){
    if(x<y){
        x=y;
        return 1;
    }
    return 0;
}
P unite(P a,P b){
    ll d=0;
    P res;
    if(upd(d,cal(a.fi,a.se)))res=P(a.fi,a.se);
    if(upd(d,cal(b.fi,b.se)))res=P(b.fi,b.se);
    if(upd(d,cal(a.fi,b.fi)))res=P(a.fi,b.fi);
    if(upd(d,cal(a.fi,b.se)))res=P(a.fi,b.se);
    if(upd(d,cal(a.se,b.fi)))res=P(a.se,b.fi);
    if(upd(d,cal(a.se,b.se)))res=P(a.se,b.se);
    return res;
}
ll ask(int l,int r){
	int len=lg2[r-l+1];
	P now=unite(ans[l][len],ans[r-(1<<len)+1][len]);
    return cal(now.fi,now.se);
}
int main(){
    scanf("%d%d",&n,&q);
    for(int i=2;i<M;++i){
        lg2[i]=lg2[i>>1]+1;
    }
    for(int i=1;i<n;++i){
        scanf("%d%d%d",&u,&v,&w);
        E[u].pb(P(v,w));
        E[v].pb(P(u,w));
    }
    dfs(1,-1,0);
    ST();
    for(int i=1;i<=n;++i){
        ans[i][0]=P(i,i);
    }
    for(int len=1;(1<<len)<=n;++len){
        for(int l=1;l+(1<<len)-1<=n;++l){
            ans[l][len]=unite(ans[l][len-1],ans[l+(1<<(len-1))][len-1]);
        }
    }
    while(q--){
        scanf("%d%d",&l,&r);
        printf("%lld\n",ask(l,r));
    }
	return 0;
}