Transit Tree Path

http://exam.upc.edu.cn/problem.php?id=6690

题意：已知一棵无向树有n个点，n-1条边，现给你m个问题和一个点k，每个问题包括两个点x和y，问从结点x到结点y经过结点k的最短路径。

思路：因为是树，所以每个点都连通。结点x和y通过结点k的最短路径可以分成两部分：结点k到x的最短路径和结点k到y的最短路径。如此，通过一个简单的DFS，求解结点k到i(1≤i≤n)的路径长度dis[i]。则查询的返回值为dis[x]+dis[y]

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
vector<pair<int ,int> >a[200005];
int vis[100005]={0};
long long int dis[100005]={0};
void dfs(int k)
{
   vis[k]=1;
   for(int i=0;i<a[k].size();i++)
   {
      int fir=a[k][i].first;
      int sec=a[k][i].second;
      if(vis[fir])
        continue;
      dis[fir]=dis[k]+sec;
      dfs(fir);
   }
}
int main()
{
   int i,j,n;
   scanf("%d",&n);
   for(i=1;i<n;i++)
   {
      int u,v;
      long long int w;
      scanf("%d%d%lld",&u,&v,&w);
      a[u].push_back(make_pair(v,w));
      a[v].push_back(make_pair(u,w));
   }
   int m,k;
   scanf("%d%d",&m,&k);
   dfs(k);
   for(i=0;i<m;i++)
   {
      int x,y;
      scanf("%d%d",&x,&y);
      printf("%lld\n",dis[x]+dis[y]);
   }
   return 0;
}