POJ 1984 Navigation Nightmare（二维带权并查集）

题目链接：http://poj.org/problem?id=1984

题目大意：有n个点，在平面上位于坐标点上，给出m关系F1  F2  L  D ，表示点F1往D方向走L距离到点F2，然后给出一系列询问F1   F2  I，表示在第I个关系给出后询问F1和F2两点间的曼哈顿距离，或者未知则输出-1。

解题思路：带权并查集，但是要开二维，val[][0]表示上下（南北）方向的偏移量，val[][1]表示左右（东西）方向的偏移量，然后一直更新就好，记得两个维度都要一起更新。

代码：

 #include<iostream>
 #include<vector>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 const int N=5e4+;
 int root[N],val[N][],res[N];//val[][]用来记录权值，val[][0]表示南北方向的偏移量，val[][1]表示东西方向的偏移量 

 struct node{
     int a,b,len;
     char det[];
 }arr[N];

 struct Node{
     int a,b,id;
     Node(){}
     Node(int l,int r,int c){
         a=l;
         b=r;
         id=c;
     }
 };
 vector<Node>v[N];

 int find(int x){
     if(root[x]==x)
         return x;
     int tmp=find(root[x]);
     val[x][]+=val[root[x]][];
     val[x][]+=val[root[x]][];
     return root[x]=tmp;
 }

 int main(){
     int n,m;
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++){
         root[i]=i;
     }
     for(int i=;i<=m;i++){
         scanf("%d%d%d%s",&arr[i].a,&arr[i].b,&arr[i].len,arr[i].det);
     }
     int q;
     scanf("%d",&q);
     for(int i=;i<=q;i++){
         int a,b,num;
         scanf("%d%d%d",&a,&b,&num);
         v[num].push_back(Node(a,b,i));
     }

     for(int i=;i<=m;i++){
         int a=arr[i].a,b=arr[i].b,len=arr[i].len;
         int t1=find(a);
         int t2=find(b);
         if(t1!=t2){
             root[t2]=t1;
             if(arr[i].det[]=='N'){
                 val[t2][]=val[a][]-val[b][]+len;//val[t2][0]+val[b][0]=val[a][0]+len即a点向北走len距离到b点
                 val[t2][]=val[a][]-val[b][];//东西方向没有发生偏移
             }
             if(arr[i].det[]=='S'){
                 val[t2][]=val[a][]-val[b][]-len;//val[t2][0]+val[b][0]=val[a][0]+len即a点向南走len距离到b点
                 val[t2][]=val[a][]-val[b][];
             }
             if(arr[i].det[]=='W'){
                 val[t2][]=val[a][]-val[b][]-len;//同理
                 val[t2][]=val[a][]-val[b][];
             }
             if(arr[i].det[]=='E'){
                 val[t2][]=val[a][]-val[b][]+len;
                 val[t2][]=val[a][]-val[b][];
             }
         }
         for(int j=;j<v[i].size();j++){
             a=v[i][j].a;
             b=v[i][j].b;
             t1=find(a);
             t2=find(b);
             if(t1!=t2)//两点不在同一个并查集中则两点间没有通路
                 res[v[i][j].id]=-;
             else
                 res[v[i][j].id]=abs(val[a][]-val[b][])+abs(val[a][]-val[b][]);
         }
     }
     for(int i=;i<=q;i++){
         printf("%d\n",res[i]);
     }
     return ;
 }