【jzoj3824】【codeforces RCC 2014 Warmup (Div. 1) D】【Big Problems for Organizers】【树】

题目大意

给出一棵树，每次询问两个点，任意一个点到这两个点距离最小值最大是多少。

解题思路

首先我们发现最远的点对一定经过直径，不然直径就不是这一条了。先找出直径，求出每个点在哪个点接入直径，直径上的点到接入它的最远点最远是多少。

求出询问的两个点在直径上的中点是那个，答案可能是直径端点到询问的两个点的距离，也可能是询问的两个点到它控制的范围内的最远点，中点的左右，这个我们可以用rmq来做。可以证明答案一定在考虑的范围内。

由于我打得头都晕了，代码很丑。

code

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define min(a,b) ((a<b)?a:b)
#define max(a,b) ((a>b)?a:b)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=2*1e5,inf=1e9;
int n,m,uu,vv,mx,gra,lg2,to[maxn*2+10],next[maxn*2+10],begin[maxn+10],
dep[maxn+10],deppest[maxn+10],fa[maxn+10][30],a[maxn+10],b[maxn+10],
vis[maxn+10],bel[maxn+10],f[maxn+10][50],g[maxn+10][50],pow2[50];
void insert(int u,int v){
    to[++gra]=v;
    next[gra]=begin[u];
    begin[u]=gra;
}
void dfs(int now,int pre){
    int far=0,far2=0,farbel=now,far2bel=now;
    for(int i=begin[now];i;i=next[i])if(to[i]!=pre){
        dep[to[i]]=dep[now]+1;
        dfs(to[i],now);
        fa[to[i]][0]=now;
        if(dep[deppest[to[i]]]-dep[now]>far){
            far=dep[deppest[to[i]]]-dep[now];
            farbel=to[i];
        }
    }
    for(int i=begin[now];i;i=next[i])if((to[i]!=pre)&&(to[i]!=farbel)&&(dep[deppest[to[i]]]-dep[now]>far2)){
        far2=dep[deppest[to[i]]]-dep[now];
        far2bel=to[i];
    }
    deppest[now]=now;
    if(far+far2>mx){
        mx=far+far2;
        uu=deppest[farbel];
        vv=deppest[far2bel];
    }
    deppest[now]=deppest[farbel];
}
int lc(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    fd(i,lg2,0)if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
    if(x==y)return x;
    fd(i,lg2,0)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int count(int u,int v){
    return dep[u]+dep[v]-2*dep[lc(u,v)];
}
void dfss(int now,int col,int di){
    bel[now]=col;b[col]=max(b[col],di);vis[now]=1;
    for(int i=begin[now];i;i=next[i])if(!vis[to[i]])dfss(to[i],col,di+1);
    vis[now]=0;
}
int countt(int x,int y){
    int lg2=log(y-x+1)/log(2);
    return max(f[x][lg2],f[y-pow2[lg2]+1][lg2]);
}
int countt2(int x,int y){
    int lg2=log(y-x+1)/log(2);
    return max(g[x][lg2],g[y-pow2[lg2]+1][lg2]);
}
int main(){
    //freopen("d.in","r",stdin);
    //freopen("d.out","w",stdout);
    scanf("%d",&n);int u,v;
    fo(i,2,n){
        scanf("%d%d",&u,&v);
        insert(u,v);insert(v,u);
    }
    scanf("%d",&m);
    lg2=log(n)/log(2);
    dep[1]=1;dfs(1,0);
    fo(j,1,lg2)fo(i,1,n)fa[i][j]=fa[fa[i][j-1]][j-1];
    a[a[0]=1]=uu;int lca=lc(uu,vv),len=count(uu,vv)+1;
    while(a[a[0]]!=lca)a[a[0]+1]=fa[a[a[0]]][0],a[0]++;
    a[len]=vv;
    fd(i,len-1,a[0]+1)a[i]=fa[a[i+1]][0];
    fo(i,1,len){
        vis[a[i-1]]=vis[a[i+1]]=1;
        dfss(a[i],i,0);
        vis[a[i-1]]=vis[a[i+1]]=0;
    }
    fo(i,1,len)f[i][0]=b[i]+i;
    fo(i,1,len)g[i][0]=b[i]+n-i+1;
    pow2[0]=1;fo(i,1,lg2)pow2[i]=pow2[i-1]*2;
    fo(j,1,lg2)fo(i,1,len)f[i][j]=max(f[i][j-1],f[i+pow2[j-1]][j-1]),
                          g[i][j]=max(g[i][j-1],g[i+pow2[j-1]][j-1]);
    fo(i,1,m){
        scanf("%d%d",&u,&v);
        int a1=max(count(u,v)/2,max(min(count(uu,u),count(uu,v)),min(count(vv,u),count(vv,v))));
        if(bel[u]>bel[v])swap(u,v);
        int a2=bel[u]+count(u,v)/2-count(u,a[bel[u]]);
        if(bel[u]!=bel[v]){
            if(bel[u]+1<=a2)a1=max(a1,countt(bel[u]+1,a2)-bel[u]+count(u,a[bel[u]]));
            if(a2+1<=bel[v]-1)a1=max(a1,countt2(a2+1,bel[v]-1)-(n-bel[v]+1)+count(v,a[bel[v]]));
        }else a1=max(a1,b[bel[u]]-max(count(u,a[bel[u]]),count(v,a[bel[v]])));
        printf("%d\n",a1);
    }
    return 0;
}