並查集練習---poj 1984

usaco的月賽題。

記錄兩個點之間x方向和y方向的相對距離，用並查集維護。

若與poj 1182食物鏈進行比較，便會發現路徑壓縮部分，集合合並部分的相似點。

所以並查集不難，是有一定套路可循的。大家一定要好好總結。

【代碼】

[cpp] 
#include <iostream> 
#include <cstring> 
#include <string> 
#include <cstdio> 
#include <algorithm> 
using namespace std; 
 
const int N=40005,K=10005; 
 
struct node 
{ 
    int x,y,idx,n; 
}a[K]; 
 
int x[N],y[N],dx[N],dy[N],rx[N],ry[N],ans[K],fa[N]; 
int n,m,k; 
 
int find(int x) 
{ 
    if (fa[x]==x) return x; 
    int t=fa[x]; 
    fa[x]=find(fa[x]); 
    rx[x]+=rx[t]; 
    ry[x]+=ry[t]; 
    return fa[x]; 
} 
 
int cmp(node a,node b) 
{ 
    return a.idx<b.idx; 
} 
 
int main() 
{ 
    int i,j,d,fx,fy; 
    char c; 
 
    freopen("in","r",stdin); 
    scanf("%d%d",&n,&m); 
    for (i=1;i<=n;i++) 
    { 
        fa[i]=i; 
        rx[i]=ry[i]=0; 
    } 
    for (i=1;i<=m;i++) 
    { 
        scanf("%d%d%d %c",&x[i],&y[i],&d,&c); 
        switch(c) 
        { 
            case 'W':dx[i]=-d;dy[i]=0;break; 
            case 'S':dx[i]=0;dy[i]=-d;break; 
            case 'E':dx[i]=d;dy[i]=0;break; 
            case 'N':dx[i]=0;dy[i]=d;break; 
        } 
    } 
    scanf("%d",&k); 
    for (i=1;i<=k;i++) 
    { 
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].idx); 
        a[i].n=i; 
    } 
    sort(a+1,a+k+1,cmp); 
    j=1; 
    for (i=1;i<=k;i++) 
    { 
        for (;j<=a[i].idx;j++) 
        { 
            fx=find(x[j]);fy=find(y[j]); 
            fa[fy]=fx; 
            rx[fy]=rx[x[j]]-rx[y[j]]-dx[j]; 
            ry[fy]=ry[x[j]]-ry[y[j]]-dy[j]; 
        } 
        if (find(a[i].x)!=find(a[i].y)) 
            ans[a[i].n]=-1; 
        else 
            ans[a[i].n]=abs(rx[a[i].x]-rx[a[i].y])+abs(ry[a[i].x]-ry[a[i].y]); 
    } 
    for (i=1;i<=k;i++) 
        printf("%d\n",ans[i]); 
}