BZOJ_3362_[Usaco2004 Feb]Navigation Nightmare 导航噩梦_并查集

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 40050
int n,fa[N],xx[N],yy[N],m,ans[N];
char opt[10];
int Abs(int x) {
    return x>0?x:-x;
}
struct E {
    int a,b,c,d;
    void rd() {
        scanf("%d%d%d%s",&a,&b,&c,opt);
        if(opt[0]==‘N‘) d=1;
        if(opt[0]==‘S‘) d=2;
        if(opt[0]==‘W‘) d=3;
        if(opt[0]==‘E‘) d=4;
    }
}e[N];
struct Q {
    int t,x,y,id;
    void rd(){scanf("%d%d%d",&x,&y,&t);}
    bool operator < (const Q &u) const {
        return t<u.t;
    }
}q[N];
int find(int x) {
    if(fa[x]==x) return x;
    int tmp=fa[x];
    fa[x]=find(fa[x]);
    xx[x]+=xx[tmp];
    yy[x]+=yy[tmp];
    return fa[x];
}
void add(int x) {
    int a=e[x].a,b=e[x].b,c=e[x].c,d=e[x].d;
    int da=find(a),db=find(b);
    fa[da]=db;
    // if(ta != tb) f[ta] = tb , dx[ta] = dx[b[t]] + cx[t] - dx[a[t]] , dy[ta] = dy[b[t]] + cy[t] - dy[a[t]];
    if(d==1) {
        xx[da]=xx[b]-xx[a];
        yy[da]=yy[b]-yy[a]-c;
    }else if(d==2) {
        xx[da]=xx[b]-xx[a];
        yy[da]=yy[b]-yy[a]+c;
    }else if(d==3) {
        xx[da]=xx[b]-xx[a]-c;
        yy[da]=yy[b]-yy[a];
    }else {
        xx[da]=xx[b]-xx[a]+c;
        yy[da]=yy[b]-yy[a];
    }
}
int query(int x,int y) {
    int dx=find(x),dy=find(y);
    if(dx!=dy) return -1;
    return Abs(xx[x]-xx[y])+Abs(yy[x]-yy[y]);
}
int main() {
    scanf("%d%*d",&n);
    int i;
    for(i=1;i<=n;i++) fa[i]=i;
    for(i=1;i<n;i++) {
        e[i].rd();
    }
    scanf("%d",&m);
    for(i=1;i<=m;i++) {
        q[i].rd(); q[i].id=i;
    }
    sort(q+1,q+m+1);
    int now=1;
    for(i=1;i<=m;i++) {
        while(now<=q[i].t) {
            add(now); now++;
        }
        ans[q[i].id]=query(q[i].x,q[i].y);
    }
    for(i=1;i<=m;i++) {
        printf("%d\n",ans[i]);
    }
}