标签：return 线段 mid Ask gdis pair 51nod1766 ll 最远
正题
题目链接:http://www.51nod.com/Challenge/Problem.html#problemId=1766
题目大意
给出 n n n个点的一棵树， m m m次询问给出两个区间，要求在两个区间中各选一个点使得他们之间距离最大。
1 ≤ n , m ≤ 1 0 5 1\leq n,m\leq 10^5 1≤n,m≤105
解题思路
结论就是两个区间中选择的点都是在各自区间中距离最远的两个点中的一个。
证明：假设集合中最长的为 l e n len len，设两端的点为 x , y x,y x,y。如果走到点 x x x不是最远的那么走到 y y y肯定是最远的，因为如果存在更远的点在集合内那么显然 x x x走到那个点是比 x x x走到 y y y远的。
考虑如何快速询问区间的直径端点，对于两个相邻的区间 [ l , m i d ] , [ m i d + 1 , r ] [l,mid],[mid+1,r] [l,mid],[mid+1,r]，我们可以用上述的方法合并得到区间 [ l , r ] [l,r] [l,r]的答案。
这启示我们可以用线段树，用 R M Q RMQ RMQ预处理树上距离即可。
时间复杂度： O ( n log n ) O(n\log n) O(nlogn)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mp(x,y) make_pair(x,y)
#define ll long long
using namespace std;
const ll N=2e5+10;
struct node{
ll to,next,w;
}a[N<<1];
ll n,q,tot,cnt,f[N][19],lg[N];
ll ls[N],dis[N],dep[N],rfn[N];
pair<ll,ll> w[N<<1];
void addl(ll x,ll y,ll w){
a[++tot].to=y;
a[tot].next=ls[x];
a[tot].w=w;ls[x]=tot;
return;
}
void dfs(ll x,ll fa){
f[++cnt][0]=x;rfn[x]=cnt;
dep[x]=dep[fa]+1;
for(ll i=ls[x];i;i=a[i].next){
ll y=a[i].to;
if(y==fa)continue;
dis[y]=dis[x]+a[i].w;
dfs(y,x);f[++cnt][0]=x;
}
return;
}
ll LCA(ll x,ll y){
ll l=rfn[x],r=rfn[y];
if(l>r)swap(l,r);
ll z=lg[r-l+1];
x=f[l][z];y=f[r-(1<<z)+1][z];
return (dep[x]<dep[y])?x:y;
}
ll gdis(ll x,ll y)
{return dis[x]+dis[y]-dis[LCA(x,y)]*2;}
pair<ll,ll> Merge(const pair<ll,ll> &a,const pair<ll,ll> &b){
ll p[4]={a.first,a.second,b.first,b.second};
pair<ll,ll> c=mp(p[0],p[1]);ll w=gdis(p[0],p[1]);
for(ll i=0;i<4;i++)
for(ll j=i+1;j<4;j++){
if(!i&&j==1)continue;
ll k=gdis(p[i],p[j]);
if(k>w)w=k,c=mp(p[i],p[j]);
}
return c;
}
void Build(ll x,ll L,ll R){
if(L==R){w[x]=mp(L,L);return;}
ll mid=(L+R)>>1;
Build(x*2,L,mid);
Build(x*2+1,mid+1,R);
w[x]=Merge(w[x*2],w[x*2+1]);
return;
}
pair<ll,ll> Ask(ll x,ll L,ll R,ll l,ll r){
if(L==l&&R==r)return w[x];
ll mid=(L+R)>>1;
if(r<=mid)return Ask(x*2,L,mid,l,r);
if(l>mid)return Ask(x*2+1,mid+1,R,l,r);
return Merge(Ask(x*2,L,mid,l,mid),Ask(x*2+1,mid+1,R,mid+1,r));
}
signed main()
{
scanf("%lld",&n);
for(ll i=1;i<n;i++){
ll x,y,w;
scanf("%lld%lld%lld",&x,&y,&w);
addl(x,y,w);addl(y,x,w);
}
dfs(1,0);
for(ll j=1;(1<<j)<=cnt;j++)
for(ll i=1;i+(1<<j)-1<=cnt;i++){
ll x=f[i][j-1],y=f[i+(1<<j-1)][j-1];
f[i][j]=(dep[x]<dep[y])?x:y;
}
for(ll i=2;i<=cnt;i++)lg[i]=lg[i>>1]+1;
Build(1,1,n);
scanf("%lld",&q);
while(q--){
ll l1,r1,l2,r2;
scanf("%lld%lld%lld%lld",&l1,&r1,&l2,&r2);
pair<ll,ll> a,b;
a=Ask(1,1,n,l1,r1);
b=Ask(1,1,n,l2,r2);
ll x=a.first,y=a.second;
ll l=b.first,r=b.second;
printf("%lld\n",max(max(gdis(x,l),gdis(x,r)),max(gdis(y,l),gdis(y,r))));
}
return 0;
}
标签：return,线段,mid,Ask,gdis,pair,51nod1766,ll,最远 来源： https://blog.csdn.net/Mr_wuyongcong/article/details/121195998
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。