[Usaco2008 Oct]牧场行走
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)对奶牛之间的距离。
土地被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,表示两头奶牛的编号。
*第二行到第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
题解:非常裸的一道LCA题。。(怎么又是这句话)。。。只要从[Poi0202]Travelling Salesman 最近公共祖先 这道题的基础上改一点就好了。。
不信的看这里:
[Poi0202]Travelling Salesman 最近公共祖先
附上代码:
#include<bits/stdc++.h>
using namespace std;
int n,q,cnt;
int deep[1001],head[1001],dis[1001],fa[1001][11];
bool vis[1001];
struct Edge{int to,next,v;} edge[2001];
int read()
{
int x=0,f=1;
char ch=getchar();
while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void ins(int u,int v,int w){edge[++cnt].to=v;edge[cnt].next=head[u];edge[cnt].v=w;head[u]=cnt;}
void getf(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];
}
}
void dfs(int x)
{
getf(x);
for(int i=head[x];i;i=edge[i].next)
{
if(vis[edge[i].to])continue;
deep[edge[i].to]=deep[x]+1;
dis[edge[i].to]=dis[x]+edge[i].v;
fa[edge[i].to][0]=x;
dfs(edge[i].to);
}
}
int gets(int a,int b)
{
for(int i=0;i<=9;i++)
if(b&(1<<i))a=fa[a][i];
return a;
}
int lca(int u,int v)
{
if(deep[u]<deep[v])swap(u,v);
u=gets(u,deep[u]-deep[v]);
if(u==v)return u;
for(int i=9;i>=0;i--)
if(fa[u][i]!=fa[v][i])
u=fa[u][i],v=fa[v][i];
return fa[u][0];
}
int main()
{
n=read();q=read();
for(int i=1,u,v,w; i<n; i++)
{
u=read();v=read();w=read();
ins(u,v,w),ins(v,u,w);
}
dfs(1);
for(int i=1,x,y;i<=q;i++)
{
x=read();y=read();
printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
}
}