题目
牛妹种了一棵苹果树。
这棵苹果树有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;
}