bzoj 1602: [Usaco2008 Oct]牧场行走

1602: [Usaco2008 Oct]牧场行走

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 1603   Solved: 826
[ Submit][ Status][ Discuss]

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

4 2
2 1 2
4 3 2
1 4 3
1 2
3 2

Sample Output

2
7

HINT

Source

资格赛

#include<iostream>  
#include<cstdio>  
#include<algorithm>  
#include<cstring>  
#define N 100000   
using namespace std;  
int n,m;  
int next[N],point[N],v[N],c[N],tot,deep[N],mi[20],fa[1003][20]; 
int len[1003][20]; 
void add(int x,int y,int z)
{
  tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
  tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void build(int x,int f,int depth)
{
  deep[x]=depth;
  for (int i=1;i<=10;i++)
  {
   if (deep[x]-mi[i]<=0) break;
   fa[x][i]=fa[fa[x][i-1]][i-1];
   len[x][i]=len[x][i-1]+len[fa[x][i-1]][i-1];
  }
  for (int i=point[x];i;i=next[i])
  if (v[i]!=f)
  {
   fa[v[i]][0]=x;
   len[v[i]][0]=c[i];
   build(v[i],x,depth+1);
  }
}
int lca(int x,int y)
{
 if (deep[x]<deep[y]) swap(x,y);
 int k=deep[x]-deep[y]; int sum=0;
 for (int i=0;i<=10;i++)
 if ((k>>i)&1)  sum+=len[x][i],x=fa[x][i];
 if (x==y) return sum;
 for (int i=10;i>=0;i--)
  if (fa[x][i]!=fa[y][i])
  {
   sum+=len[x][i];
   sum+=len[y][i];
   x=fa[x][i];
   y=fa[y][i];
  }
 sum+=len[x][0];
 sum+=len[y][0];
 return sum;
}
int main()  
{  
  scanf("%d%d",&n,&m);  
  for (int i=1;i<=n-1;i++)  
   {  
    int x,y,z; scanf("%d%d%d",&x,&y,&z);  
    add(x,y,z);
   }  
  mi[0]=1;
  for (int i=1;i<=10;i++)
   mi[i]=mi[i-1]*2;
  build(1,0,1);
  for (int i=1;i<=m;i++)
  {
   int x,y; scanf("%d%d",&x,&y);
   printf("%d\n",lca(x,y));
  }
}