追捕游戏 LCA

NKOJ4239 追捕游戏

问题描述

何老板最近在玩一款追铺游戏，游戏虽然简单，何老板仍旧乐此不疲。
游戏地图中有n座城市由n-1条双向道路连接。任意两座城市都可相互到达。一名罪犯从A城市出发沿最短路线逃往B城市。在罪犯出发的同时，何老板控制一名警察从C城市出发去追捕那名罪犯。每条道路都有一定的长度(单位米)。罪犯和警察行走的速度相同，都是1秒钟行走1米。

若罪犯到达B城市时还没有被抓住，何老板就输掉了这局游戏。何老板总共玩了m局游戏，每局游戏开始前，何老板想知道他是否能赢下这局游戏，如果能，警察最少行走多少米才能抓到罪犯？

输入格式

第一行，两个整数n和m

接下来n-1行，每行三个整数X,Y,Z，表示城市X和Y之间有一条长度为Z的道路相连。
接下来m行，每行三个整数A,B,C。

输出格式

m行，每行对应一局游戏的结果。若能抓捕到罪犯。输出一个整数，表示警察最少需要行走的距离。若无法抓到罪犯，输出-1。

样例输入

11 2
1 2 6
1 3 3
1 4 3
3 5 2
3 6 5
4 7 9
6 10 3
5 8 4
5 9 3
8 11 8
11 9 10
2 4 7

样例输出

10
9

样例解释

第1局在5号城市抓住罪犯。
第2局在4号城市抓住罪犯。

数据范围

对于约40% 的数据：1<=N,M<=2000

对于100% 的数据：1<=N,M<=100000 1<=道路的长度<=10000

---

由于树的性质，两点之间有且仅有一条路径。因此当C不在A、B的的路径上时，C要到A、B的路径上，只有一个唯一的点D进行中转（否则就会有环）。也就是说，警察的路径也是唯一的。那么只需要比较警察和小偷到达D点的时间即可。

通常的想法是，先找出D的位置，再计算dis(A,D)和dis(C,D)。然而这样实现起来比较麻烦。事实上，容易得到以下关系：

dis(A,D)=dis(A,C)+dis(A,B)-2*dis(B,C)
dis(C,D)=dis(A,C)+dis(B,C)-2*dis(A,B)

快速查找树上两点的距离，倍增即可。

---

#include<stdio.h>
#define Min(x,y) ((x<y)?(x):(y))
#define MAXM 200005
#define MAXN 100005

int N,M,Dis;

int en[MAXM],las[MAXN],nex[MAXM],len[MAXM],tot;
void ADD(int x,int y,int z)
{
    en[++tot]=y;
    nex[tot]=las[x];
    las[x]=tot;
    len[tot]=z;
}

int dep[MAXN],fa[MAXN][20],dis[MAXN][20];
void DFS(int x,int f,int z)
{
    int i,y;

    dep[x]=dep[f]+1;
    fa[x][0]=f;dis[x][0]=z;
    for(i=1;i<20;i++)fa[x][i]=fa[fa[x][i-1]][i-1],dis[x][i]=dis[x][i-1]+dis[fa[x][i-1]][i-1];

    for(i=las[x];i;i=nex[i])
    {
        y=en[i];
        if(y!=f)DFS(y,x,len[i]);
    }
}

int LCA(int x,int y)
{
    int i,t,d;
    Dis=0;

    if(dep[x]<dep[y])t=x,x=y,y=t;
    d=dep[x]-dep[y];
    for(i=0;i<20;i++)if(d&(1<<i))Dis+=dis[x][i],x=fa[x][i];
    if(x==y)return x;
    for(i=19;i>=0;i--)if(fa[x][i]!=fa[y][i])Dis+=dis[x][i]+dis[y][i],x=fa[x][i],y=fa[y][i];
    Dis+=dis[x][0]+dis[y][0];
    return fa[x][0];
}

int main()
{
    int i,x,y,z,A,B,C,police,criminal;

    scanf("%d%d",&N,&M);
    for(i=1;i<N;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        ADD(x,y,z);
        ADD(y,x,z);
    }

    DFS(1,0,0);

    for(i=1;i<=M;i++)
    {
        scanf("%d%d%d",&A,&B,&C);
        LCA(B,C);x=Dis;
        LCA(C,A);y=Dis;
        LCA(A,B);z=Dis; 

        police=x+y-z>>1;
        criminal=y+z-x>>1;

        if(police>criminal)puts("-1");
        else printf("%d\n",police);
    }
}

---

小小总结一下：
树上的一些问题，鉴于树上两点间有且仅有一条路径，仅使用距离讨论就可以简单而方便的解决很多问题。