Navigation Nightmare（并查集（多权值更新））

题目来源：https://vjudge.net/problem/POJ-1984
【题意】
农夫约翰有 N 个农场，标号 1 到 N，M 条的不同的垂直或水平的道路连接着农场，道路的长度不超过 1000。
【思路】
权值分为x，y轴方向上，更新当前节点到根节点的权值。
【代码】

#include<map>
#include<stack>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long LL;
const int N1=40010;
const int N2=10010;
int pre[N1];
int rx[N1],ry[N1];
struct p1
{
    int x1,x2,dis;
    char s;
} st[N1];
struct p2
{
    int x,y,d,num;
    friend bool operator<(const p2&a,const p2&b)
    {
        return a.d<b.d;
    }
} query[N2];
int sum[N2];
int n,m;
int find(int x)
{
    if(pre[x]!=x)
    {
        int fx=find(pre[x]);
        rx[x]+=rx[pre[x]];
        ry[x]+=ry[pre[x]];
        pre[x]=fx;
    }
    return pre[x];
}
int cmp(int aq,int bq)
{
    return query[aq].num<query[bq].num;
}
int main()
{

    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1; i<=n; i++)
            pre[i]=i,
                   rx[i]=ry[i]=0;
        for(int i=1; i<=m; i++)
            scanf("%d%d%d%*c%c",&st[i].x1,&st[i].x2,&st[i].dis,&st[i].s);
        int k;
        scanf("%d",&k);
        for(int i=0; i<k; i++)
            scanf("%d%d%d",&query[i].x,&query[i].y,&query[i].d),query[i].num=i;
        sort(query,query+k);
        int j=1;
        for(int i=0; i<k; i++)
        {
            for(; j<=query[i].d; j++)
            {
                int fx1=find(st[j].x1),fx2=find(st[j].x2);
                if(fx1!=fx2)
                {
                    pre[fx2]=fx1;
                    if(st[j].s=='E') rx[fx2]=rx[st[j].x1]+st[j].dis-rx[st[j].x2],ry[fx2]=ry[st[j].x1]-ry[st[j].x2];
                    else if(st[j].s=='W') rx[fx2]=rx[st[j].x1]-st[j].dis-rx[st[j].x2],ry[fx2]=ry[st[j].x1]-ry[st[j].x2];
                    else if(st[j].s=='N') ry[fx2]=ry[st[j].x1]+st[j].dis-ry[st[j].x2],rx[fx2]=rx[st[j].x1]-rx[st[j].x2];
                    else if(st[j].s=='S') ry[fx2]=ry[st[j].x1]-st[j].dis-ry[st[j].x2],rx[fx2]=rx[st[j].x1]-rx[st[j].x2];
                }
            }
            int fxa=find(query[i].x);
            int fxb=find(query[i].y);
            sum[query[i].num]=fxa==fxb?(abs(rx[query[i].x]-rx[query[i].y])+abs(ry[query[i].x]-ry[query[i].y])):-1;
        }
        for(int i=0; i<k; i++)
            printf("%d\n",sum[i]);
    }

}