bzoj1832: [AHOI2008]聚会(倍增lca)

来源:互联网 发布:紫牛软件 编辑:程序博客网 时间:2024/05/29 16:59

题目传送门
水题呀。

解法:
没学过倍增lca看这里
刚做完上一题毒瘤题scy就说初三的尽量不要看题解呀。
然后有点担忧。有些题不看题解貌似做不了诶。
然后随便点到了这道题。
看了一会儿。
卧槽?!
大水题?!
想了一会儿。
不就。。
三个点分三种情况么。肯定不能走别人走过的路呀。
然后用lca维护一下就可以了呀。
比如abc三个点,
如果a和b肯定要经过他们的最近公共祖先呀。
然后他们两个到一起了肯定是让c来他们这里而不是他们去c那里因为是双倍距离嘛。

代码实现:

#include<queue>#include<cmath>#include<cstdio>#include<cstring>#include<cstdlib>#include<iostream>#include<algorithm>using namespace std;struct node {int x,y,next;}a[1110000];int len,last[510000];void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}int mn[21][510000],dep[510000],bin[21];void dfs(int x) {    for(int k=last[x];k;k=a[k].next) {        int y=a[k].y;if(y!=mn[0][x]){mn[0][y]=x;dep[y]=dep[x]+1;dfs(y);}    }}int n;void work() {    bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;    for(int j=1;j<=20;j++)for(int i=1;i<=n;i++)if(dep[i]>=bin[j])mn[j][i]=mn[j-1][mn[j-1][i]];}int lca(int x,int y) {    if(dep[x]>dep[y])swap(x,y);    for(int i=20;i>=0;i--)if(dep[y]-dep[x]>=bin[i]) y=mn[i][y];    if(x==y)return x;    for(int i=20;i>=0;i--)if(dep[x]>=bin[i]&&mn[i][x]!=mn[i][y]) {x=mn[i][x];y=mn[i][y];}    return mn[0][x];}int dis(int x,int y) {return dep[x]+dep[y]-2*dep[lca(x,y)];}int main() {    int m;scanf("%d%d",&n,&m);    len=0;memset(last,0,sizeof(last));    for(int i=1;i<n;i++) {int x,y;scanf("%d%d",&x,&y);ins(x,y);ins(y,x);}    mn[0][1]=0;dep[1]=0;dfs(1);work();    for(int i=1;i<=m;i++) {        int a,b,c;scanf("%d%d%d",&a,&b,&c);        int Lca_ab,Lca_ac,Lca_bc,A,B,C;        Lca_ab=lca(a,b);A=dis(a,b)+dis(Lca_ab,c);        Lca_ac=lca(a,c);B=dis(a,c)+dis(Lca_ac,b);        Lca_bc=lca(b,c);C=dis(b,c)+dis(Lca_bc,a);        if(A<B&&A<C)printf("%d %d",Lca_ab,A);        else if(B<A&&B<C)printf("%d %d",Lca_ac,B);        else printf("%d %d",Lca_bc,C);        printf("\n");    }    return 0;}