Description

N頭牛（2<=n<=1000）別人被標記為1到n，在同樣被標記1到n的n塊土地上吃草，第i頭牛在第i塊牧場吃草。 這n塊土地被n-1條邊連接。 奶牛可以在邊上行走，第i條邊連接第Ai，Bi塊牧場，第i條邊的長度Li（1<=Li<=10000）。 這些邊被安排成任意兩頭奶牛都可以通過這些邊到達的情況，所以說這是一棵樹。 這些奶牛是非常喜歡交際的，經常會去互相訪問,他們想讓你去幫助他們計算Q(1<=q<=1000)對奶牛之間的距離。

Input

*第一行：兩個被空格隔開的整數：N和Q

*第二行到第n行：第i+1行有兩個被空格隔開的整數：AI，BI，LI

*第n+1行到n+Q行：每一行有兩個空格隔開的整數：P1，P2，表示兩頭奶牛的編號。

Output

*第1行到第Q行：每行輸出一個數，表示那兩頭奶牛之間的距離。

Sample Input

Sample Output

題目鏈接：

　　　　http://www.lydsy.com/JudgeOnline/problem.php?id=1602

Solution

　　　　LCA的模板題。。。

代碼

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#define N 1010
using namespace std;
inline int Read(){
    int x=0,f=1;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}
int n,Q,cnt=0;
int hed[N];
int deep[N],dis[N],fa[N][15];
bool vis[N];
struct edge{
    int r,nxt,v;
}e[N<<1];
void insert(int u,int v,int w){
    e[++cnt].r=v;e[cnt].v=w;e[cnt].nxt=hed[u];hed[u]=cnt;
    e[++cnt].r=u;e[cnt].v=w;e[cnt].nxt=hed[v];hed[v]=cnt;
}
void dfs(int x){
    vis[x]=1;
    for(int i=1;i<=9;i++){
        if(deep[x]<(1<<i)) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=hed[x];i;i=e[i].nxt)
        if(!vis[e[i].r]){
            deep[e[i].r]=deep[x]+1;
            dis[e[i].r]=dis[x]+e[i].v;
            fa[e[i].r][0]=x;
            dfs(e[i].r);
        }
}
int LCA(int x,int y){
    if(deep[x]<deep[y]) swap(x,y);
    int d=deep[x]-deep[y];
    for(int i=0;i<=9;i++)
        if((1<<i)&d) x=fa[x][i];
    for(int i=9;i>=0;i--)
        if(fa[x][i]!=fa[y][i]){
            x=fa[x][i];
            y=fa[y][i];
        }
    if(x==y) return x;
    return fa[x][0];
}
int main(){
    int u,v,w;
    n=Read();Q=Read();
    for(int i=1;i<n;i++){
        u=Read();v=Read();w=Read();
        insert(u,v,w);
    }
    dfs(1);
    for(int i=1;i<=Q;i++){
        u=Read();v=Read();
        printf("%d\n",dis[u]+dis[v]-2*dis[LCA(u,v)]);
    }
    return 0;
}

This passage is made by Iscream-2001.

BZOJ 1602--牧場行走(LCA)