HDU 2586 查找两点之间的距离

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<math.h>
using namespace std;
#define LL long long
#define INF 0x3f3f3f3f
#define N 40006
int k,y,w[N],t;
int aa[N<<1];
struct node
{
    int f,g,v;
}a[N<<1];
void add(int e,int f,int g)
{
    a[k].f=f;
    a[k].g=g;
    a[k].v=aa[e];
    aa[e]=k++;
}
void bfs(int x,int sum)
{
    if(x==y)
        t=sum;
    if(t!=-1) return ;
    for(int i=aa[x];i!=-1;i=a[i].v)
    {
        int f=a[i].f;
        if(!w[f])
        {
            w[f]=1;
            bfs(f,sum+a[i].g);
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        k=0;
        int n,m,e,f,g;
        memset(aa,-1,sizeof(aa));
        scanf("%d%d",&n,&m);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&e,&f,&g);
            add(e,f,g);
            add(f,e,g);
        }
        while(m--)
        {
            int x;
            t=-1;
            scanf("%d%d",&x,&y);
            memset(w,0,sizeof(w));
            w[x]=1;
            bfs(x,0);
            printf("%d\n",t);
        }
    }
    return 0;
}