并查集练习---poj 1984

usaco的月赛题。

记录两个点之间x方向和y方向的相对距离，用并查集维护。

若与poj 1182食物链进行比较，便会发现路径压缩部分，集合合并部分的相似点。

所以并查集不难，是有一定套路可循的。大家一定要好好总结。

【代码】

#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]);}