bzoj1602 [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)对奶牛之间的距离。

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

很简单的lca。但我闲着无聊打了个bfs……反正时限多

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
struct aaa{
int to,next,v;
}e[5000000];
int n,m,cnt,x,y,z;
int head[100000];
int map[1001][1001];
bool mark[100000];
void ins(int u,int v,int w)
{
e[++cnt].to=v;
e[cnt].v=w;
e[cnt].next=head[u];
head[u]=cnt;
}
void insert(int u,int v,int w)
{
ins(u,v,w);
ins(v,u,w);
}
int bfs(int from)
{
int queue[10000];
int dist[10000];
int t=0,w=1;
bool mark[10000];
memset(queue,0,sizeof(queue));
memset(mark,0,sizeof(mark));
memset(dist,0,sizeof(dist));
queue[1]=from;
mark[from]=1;
while (t<w)
{
t++;
int i=head[queue[t]];
while (i)
{
if (!mark[e[i].to])
{
queue[++w]=e[i].to;
dist[e[i].to]=dist[queue[t]]+e[i].v;
mark[e[i].to]=1;
if (e[i].to==y) return dist[y];
}
i=e[i].next;
}
}
return dist[y];
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n-1;i++)
{
cin>>x>>y>>z;
insert(x,y,z);
}
for (int i=1;i<=m;i++)
{
cin>>x>>y;
memset(mark,0,sizeof(mark));
cout<<bfs(x)<<endl;
}
}