poj 1984 Navigation Nightmare (并查集)

http://poj.org/problem?id=1984

    题目描述的如此蛋疼。。。

    关系并查集，用cx和cy数组记录点与根结点的相对位置，然后根据输入依次合并，当碰到询问时便计算距离，若不在同一集合中，即无法确定相对位置。

code:

#include<cstdio>
#include<cstring>
int p[40001], cx[40001], cy[40001] ;
int f1[10001], f2[10001], fi[10001] ;
struct edge{
    int s, e, l ;
    char d ;
}q[40001] ;
int abs(int a){
    return a>0?a:-a ;
}
void make_set(int n){
    for(int i=0; i<=n+1; i++){
        p[i] = i ;
        cx[i] = cy[i] = 0 ;
    }
}
int find_set(int x){
    int temp ;
    if(x!=p[x]){
        temp = p[x] ;
        p[x] = find_set(p[x]) ;
        cx[x] = cx[temp] + cx[x] ;
        cy[x] = cy[temp] + cy[x] ;
    }
    return p[x] ;
}
void union_set(int x){
    int a = q[x].s ;
    int b = q[x].e ;
    int pa = find_set(a) ;
    int pb = find_set(b) ;
    p[pb] = pa ;
    cx[pb] = cx[a] - cx[b] ;
    cy[pb] = cy[a] - cy[b] ;
    switch(q[x].d){
        case 'E':cx[pb] += q[x].l ;break ;
        case 'W':cx[pb] -= q[x].l ;break ;
        case 'N':cy[pb] += q[x].l ;break ;
        case 'S':cy[pb] -= q[x].l ;break ;
    }
    return ;
}
int main(){
    int n, m, k, i, j, t, x, y, z, tx, ty ;
    while(~scanf("%d%d", &n, &m)){
        make_set(n) ;
        for(i=1; i<=m; i++)
            scanf("%d %d %d %c", &q[i].s, &q[i].e, &q[i].l, &q[i].d) ;
        scanf("%d", &t) ;
        for(i=0; i<t; i++)
            scanf("%d%d%d", &f1[i], &f2[i], &fi[i]) ;
        int temp = 0 ;
        for(i=1; i<=m; i++){
            tx = find_set(q[i].s) ;
            ty = find_set(q[i].e) ;
            if(tx!=ty)
                union_set(i) ;
            while(fi[temp]==i){
                tx = find_set(f1[temp]) ;
                ty = find_set(f2[temp]) ;
                if(tx!=ty)  printf("-1\n") ;
                else{
                    tx = abs(cx[f1[temp]] - cx[f2[temp]]) ;
                    ty = abs(cy[f1[temp]] - cy[f2[temp]]) ;
                    printf("%d\n", tx + ty) ;
                }
                temp ++ ;
            }
        }
    }
    return 0 ;}