bzoj1602: [Usaco2008 Oct]牧场行走 暴力dfs

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1602
1602: [Usaco2008 Oct]牧场行走
Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 1682 Solved: 867
[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

思路：
ｄｆｓｎ^2求树上两点距离；

代码：

#include<iostream>#include<stdio.h>#include<vector>#define N 1005using namespace std;struct node{    int y,v;    node(int yy,int vv)    {        y=yy,v=vv;    }node(){}};vector<node> lin[N];int aa,bb,cc;int flag;int n,q;long long ans;void dfs(int x,int d,int fa){    if(flag) return;     if(x==bb)    {        ans=d;          flag=1;    }    for(int i=0;i<lin[x].size();i++)    {        int u=lin[x][i].y;        if(u==fa) continue;        dfs(u,d+lin[x][i].v,x);    }}int main(){    scanf("%d%d",&n,&q);    for(int i=1;i<n;i++)    {        scanf("%d%d%d",&aa,&bb,&cc);        lin[aa].push_back(node(bb,cc));        lin[bb].push_back(node(aa,cc));    }    for(int i=1;i<=q;i++)    {        flag=0;        scanf("%d%d",&aa,&bb);        ans=0;        dfs(aa,0,0);        printf("%lld\n",ans);    }}