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 1005
using 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);
    }
}