POJ-1986 Distance Queries

Distance Queries

Time Limit: 2000MS	Memory Limit: 30000K
Total Submissions: 13987	Accepted: 4924
Case Time Limit: 1000MS

Description

Farmer John's cows refused to run in his marathon since he chose a path much too long for their leisurely lifestyle. He therefore wants to find a path of a more reasonable length. The input to this problem consists of the same input as in "Navigation Nightmare",followed by a line containing a single integer K, followed by K "distance queries". Each distance query is a line of input containing two integers, giving the numbers of two farms between which FJ is interested in computing distance (measured in the length of the roads along the path between the two farms). Please answer FJ's distance queries as quickly as possible!

Input

* Lines 1..1+M: Same format as "Navigation Nightmare"

* Line 2+M: A single integer, K. 1 <= K <= 10,000

* Lines 3+M..2+M+K: Each line corresponds to a distance query and contains the indices of two farms.

Output

* Lines 1..K: For each distance query, output on a single line an integer giving the appropriate distance.

Sample Input

7 6
1 6 13 E
6 3 9 E
3 5 7 S
4 1 3 N
2 4 20 W
4 7 2 S
3
1 6
1 4
2 6

Sample Output

13
3
36

Hint

Farms 2 and 6 are 20+3+13=36 apart. 

这题目别想多，跟方向没关系，LCA。

关键在于怎么将距离整出来，我们可以有这么一个方程，

           ans[id]=pre[v].len+pre[u].len-2*pre[Find(v)].len;
用一个图来表示就是这样：


pre[v].len就是1到3的距离，
pre[u].len就是1到4的距离。
pre[Find(v)].len就是1到2的距离。
这样就很好解决了所有的问题了。

还有就是，重要的事情说三遍：并查集要压缩路径！并查集要压缩路径！并查集要压缩路径！
之前几次忘了压缩，直接TLE；一脸懵逼

附上AC代码：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
struct Node{
    int f;
    int len;
}pre[50080];
vector<pair<int ,int > >query[50050];
vector<pair <int ,int > >vec[50100];
int ans[50050],vis[50100];
int Find(int x)
{
    if(pre[x].f==x)
    {
        return x;
    }
    else
    {
        pre[x].f=Find(pre[x].f);
        return pre[x].f;
    }
}
int dfs(int u,int fa,int length)
{
    pre[u].f=u;
    pre[u].len=pre[fa].len+length;
    vis[u]=1;
    for(int i=0;i<vec[u].size();i++)
    {
        int v=vec[u][i].first;
        int l=vec[u][i].second;
        if(v==fa)   continue;
        dfs(v,u,l);
    }
    for(int i=0;i<query[u].size();i++)
    {
        int v=query[u][i].first;
        int id=query[u][i].second;
        if(vis[v]==1)
        {
             ans[id]=pre[v].len+pre[u].len-2*pre[Find(v)].len;
        }
    }
    pre[u].f=fa;
}
int main()
{
    int n,m,x,y,l;
    char c;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(vis,0,sizeof(vis));
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d %c",&x,&y,&l,&c);
            vec[x].push_back({y,l});
            vec[y].push_back({x,l});
            vis[y]=1;
        }
        int k;
        cin>>k;
        for(int i=0;i<k;i++)
        {
            scanf("%d%d",&x,&y);
            query[x].push_back({y,i});
            query[y].push_back({x,i});
        }
        for(int i=1;i<=n;i++)
        {
            if(vis[i]==0)
            {
                memset(vis,0,sizeof(vis));
                dfs(i,-1,0);
                break;
            }
        }
        for(int i=0;i<k;i++)
            cout<<ans[i]<<endl;
        memset(ans,0,sizeof(ans));
        memset(pre,0,sizeof(pre));
        for(int i=0;i<=n;i++)
        {
            vec[i].clear();
            query[i].clear();
        }
    }
    return 0;
}