51nod 1766

题意:给定一个树(10^5),m个询问(10^5),每次给定a,b,c,d,在区间[a,b]中选一个点,[c,d]选一个点,使得这两个点距离最大,输出最大距离。

题解:首先,我们有一个结论:对于一个集合的直径,如果我们将这个集合分解成两个非空集合,它的端点一定在两个非空集合的两个端这4个端点中。这非常的显然。。。

那么我们就可以做到合并两个集合,我们就可以用线段树维护每个区间的直径,就好啦,完全不用复杂的数据结构。

这道题卡时间,所以LCA要用欧拉序RMQ做。

复杂度O(nlogn)

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define N 200005
 4 inline int read(){
 5     int x=0,f=1; char a=getchar();
 6     while(a<'0' || a>'9') {if(a=='-') f=-1; a=getchar();}
 7     while(a>='0' && a<='9') x=x*10+a-'0',a=getchar();
 8     return x*f;
 9 }
10 int n,m,cnt=1,euler[2*N],dep[N],head[N],fi[N],dis[N],st[N][20],mn[N][20],Log[2*N];
11 struct edges{
12     int to,c,next;
13 }e[N];
14 struct node{
15     int x1,x2,l,r;
16 }seg[4*N];
17 struct data{
18     int x1,x2;
19 };
20 inline int getdis(int u,int v){
21     if(!u || !v) return 0;
22     u=fi[u]; v=fi[v];
23     if(u>v) swap(u,v);
24     int p,len=Log[v-u+1];
25     p=mn[u][len]<mn[v-(1<<len)+1][len]?st[u][len]:st[v-(1<<len)+1][len];
26     return dis[euler[u]]+dis[euler[v]]-2*dis[p];
27 }
28 inline void update(int x){
29     int mx=0,x1=seg[x<<1].x1,x2=seg[x<<1].x2,x3=seg[x<<1|1].x1,x4=seg[x<<1|1].x2,f;
30 if((f=getdis(x1,x2))>mx) seg[x].x1=x1,seg[x].x2=x2,mx=f; if((f=getdis(x1,x3))>mx) seg[x].x1=x1,seg[x].x2=x3,mx=f;
31 if((f=getdis(x1,x4))>mx) seg[x].x1=x1,seg[x].x2=x4,mx=f; if((f=getdis(x2,x3))>mx) seg[x].x1=x2,seg[x].x2=x3,mx=f;
32 if((f=getdis(x2,x4))>mx) seg[x].x1=x2,seg[x].x2=x4,mx=f; if((f=getdis(x3,x4))>mx) seg[x].x1=x3,seg[x].x2=x4,mx=f;
33 }
34 inline void insert(){
35     int u=read(),v=read(),c=read();
36     e[++cnt]=(edges){v,c,head[u]};head[u]=cnt;
37 }
38 void dfs(int x,int fa){
39     dep[x]=dep[fa]+1; fi[x]=euler[0]+1;
40     for(int i=head[x];i;i=e[i].next){
41         if(fa==e[i].to) continue;
42         euler[++euler[0]]=x;
43         dis[e[i].to]=dis[x]+e[i].c; dfs(e[i].to,x);
44     }
45     euler[++euler[0]]=x;
46 }
47 inline void rmq_pre(){
48     for(int i=1;i<=euler[0];i++) st[i][0]=euler[i],mn[i][0]=dep[euler[i]];
49     for(int i=1;i<=18;i++)
50         for(int j=1;j<=euler[0];j++){
51             if(j+(1<<i)-1>euler[0]) break;
52             if(mn[j][i-1]<mn[j+(1<<(i-1))][i-1]) st[j][i]=st[j][i-1],mn[j][i]=mn[j][i-1];
53             else st[j][i]=st[j+(1<<(i-1))][i-1],mn[j][i]=mn[j+(1<<(i-1))][i-1];
54         }
55     Log[0]=-1; for(int i=1;i<=euler[0];i++) Log[i]=Log[i>>1]+1;
56 }
57 void build(int l,int r,int x){
58     seg[x].l=l; seg[x].r=r;
59     if(l==r) {seg[x].x1=l; seg[x].x2=0; return;}
60     int mid=(l+r)>>1;
61     build(l,mid,x<<1);
62     build(mid+1,r,x<<1|1);
63     update(x);
64 }
65 data merge(data tmp1,data tmp2){
66     int f,mx=0,x1=tmp1.x1,x2=tmp1.x2,x3=tmp2.x1,x4=tmp2.x2;
67     data ret;
68     if((f=getdis(x1,x2))>mx) ret.x1=x1,ret.x2=x2,mx=f; if((f=getdis(x1,x3))>mx) ret.x1=x1,ret.x2=x3,mx=f;
69     if((f=getdis(x1,x4))>mx) ret.x1=x1,ret.x2=x4,mx=f; if((f=getdis(x2,x3))>mx) ret.x1=x2,ret.x2=x3,mx=f;
70     if((f=getdis(x2,x4))>mx) ret.x1=x2,ret.x2=x4,mx=f; if((f=getdis(x3,x4))>mx) ret.x1=x3,ret.x2=x4,mx=f;
71     return ret;
72 }
73 data query(int L,int R,int x){
74     int l=seg[x].l,r=seg[x].r;
75     if(l==L && r==R) {return (data){seg[x].x1,seg[x].x2};}
76     int mid=(l+r)>>1;
77     if(R<=mid) return query(L,R,x<<1);
78     else if(mid<L) return query(L,R,x<<1|1);
79     else return merge(query(L,mid,x<<1),query(mid+1,R,x<<1|1));
80 }
81 int main(){
82     n=read();
83     for(int i=1;i<n;i++) insert();
84     dfs(1,0); rmq_pre(); build(1,n,1);
85     m=read();
86     while(m--){
87         int t1,t2,a=read(),b=read(),c=read(),d=read();
88         data tmp1=query(a,b,1),tmp2=query(c,d,1);
89         t1=max(getdis(tmp1.x1,tmp2.x1),getdis(tmp1.x1,tmp2.x2));
90         t2=max(getdis(tmp1.x2,tmp2.x1),getdis(tmp1.x2,tmp2.x2));
91         printf("%d
",max(t1,t2));
92     }
93     return 0;
94 }
原文地址:https://www.cnblogs.com/enigma-aw/p/6344784.html