POJ ~ 1984 ~ Navigation Nightmare （带权并查集，多权值）

题意：有N个地点，M组关系，每组关系有三个数字a,b,d和一个字符dir。表示a在b的dir方向的d米处。然后有k次询问，每次询问

包括三个数字，a,b,times，问在第times秒（也就是只有上面关系中的前times行）的时候a到b的曼哈顿距离为多少，没有输出-1。

k次询问的times保证从小到大。

思路：带权并查集，只不过有两个权值，一个东西方向（以东为正方向），一个南北方向（以北为正方向）。因为询问是在关系给完之后才询问的，所以我们需要先把关系给存储下来。然后根据询问的times一步步建立并查集。

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
using namespace std;
const int MAXN = 40005;
const int MAXM = 40005;
const int MAXK = 10005;
int n, m, k, ans[MAXK];
struct F
{
    int pre, w[2];//父节点和两个权值(东西，南北)
}f[MAXN];
struct Rode
{
    int a, b, d;//a-b距离为d
    char dir;//方向
}rode[MAXM];
struct Query
{
    int a, b, times;
}query[MAXK];//查询
void init()
{
    query[0].times = 0;
    for (int i = 0; i <= n; i++)
    {
        f[i].pre = i;
        f[i].w[0] = 0; f[i].w[1] = 0;
    }
}
void input()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= m; i++)
    {
        scanf("%d%d%d %c", &rode[i].a, &rode[i].b, &rode[i].d, &rode[i].dir);
    }
    scanf("%d", &k);
    for (int i = 1; i <= k; i++)
    {
        scanf("%d%d%d", &query[i].a, &query[i].b, &query[i].times);
    }
}
int Find(int x)
{
    if (f[x].pre == x) return x;
    int t = f[x].pre;
    f[x].pre = Find(f[x].pre);
    f[x].w[0] = f[x].w[0] + f[t].w[0];//更新东西权值
    f[x].w[1] = f[x].w[1] + f[t].w[1];//更新南北权值
    return f[x].pre;
}
void Union(int a, int b, int d, char dir)
{
    int root1 = Find(a), root2 = Find(b);
    if (root1 != root2)
    {
        f[root1].pre = root2;
        if (dir == 'E' || dir == 'W')
        {
            if (dir == 'W') d = -d;//以东为正方向
            f[root1].w[0] = f[b].w[0] + d - f[a].w[0];//更新东西权值
            f[root1].w[1] = f[b].w[1] - f[a].w[1];//更新南北权值
        }
        else if(dir == 'N' || dir == 'S')
        {
            if (dir == 'S') d = -d;//以北为正方向
            f[root1].w[0] = f[b].w[0] - f[a].w[0];//更新东西权值
            f[root1].w[1] = f[b].w[1] + d - f[a].w[1];//更新南北权值
        }
    }
}
int main()
{
    input();//输入
    init();//初始化
    for (int i = 1; i <= k; i++)
    {
        for (int j = query[i - 1].times + 1; j <= query[i].times; j++)
        {
            Union(rode[j].a, rode[j].b, rode[j].d, rode[j].dir);
        }
        int a = query[i].a, b = query[i].b;
        int root1 = Find(a), root2 = Find(b);
        if (root1 == root2)
        {
            ans[i] = abs(f[a].w[0] - f[b].w[0]) + abs(f[a].w[1] - f[b].w[1]);
        }
        else ans[i] = -1;
    }
    for (int i = 1; i <= k; i++) printf("%d\n", ans[i]);//输出答案
    return 0;
}
/*
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
1 4 3
2 6 6
*/