哈理工oj 1418夏夜星空并查集解题报告

这道题和poj的食物链很像，要在路径压缩的过程中进行坐标的位置处理，要保存的是当前节点相对于父节点的相对坐标位置。合并的时候要考虑清楚两个父节点之间的坐标关系是什么

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 40005
using namespace std;
struct node
{
    int p;
    int x,y;
    int r;
    int xd,yd;
};
int n;
node f[N];
void init()
{
    int i;
    for(i=1;i<=n;i++)
    {
        f[i].p=i;
        f[i].x=f[i].y=f[i].r=f[i].xd=f[i].yd=0;
    }
}
int find(int i)
{
    if(i==f[i].p)
    return i;
    int temp=f[i].p;
    f[i].p=find(temp);
    f[i].x=f[temp].x+f[i].xd;//路径压缩时进行坐标位置处理，xd,yd为相对父节点的坐标位置
    f[i].y=f[temp].y+f[i].yd;
    f[i].xd=f[i].x;
    f[i].yd=f[i].y;
    return f[i].p;
}
void uni(int i,int j,int dir,int len)
{
    int t1=find(i);
    int t2=find(j);
    //printf("%d %d\n",dir,len);
    f[t2].p=t1;
    if(dir==1)//合并时两个父节点的相对位置
    {
        f[t2].xd=f[i].x-f[j].x;
        f[t2].yd=f[i].y+len-f[j].y;
    }
    else if(dir==2)
    {
        f[t2].xd=f[i].x-f[j].x;
        f[t2].yd=f[i].y-len-f[j].y;
    }
    else if(dir==3)
    {
        f[t2].xd=f[i].x+len-f[j].x;
        f[t2].yd=f[i].y-f[j].y;
    }
    else
    {
        f[t2].xd=f[i].x-len-f[j].x;
        f[t2].yd=f[i].y-f[j].y;
    }
}
int main()
{
    int s[26];
    s['U'-'A']=1;
    s['D'-'A']=2;
    s['R'-'A']=3;
    s['L'-'A']=4;
    char s2[100];
    char s3[10];
    int m,i,j,a,b,len;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        while(m--)
        {
            scanf("%s",s2);
            if(s2[0]=='Q')
            {
                scanf("%d %d",&a,&b);
                if(find(a)==find(b))
                printf("%d\n",abs(f[a].x-f[b].x)+abs(f[a].y-f[b].y));
                else
                printf("-1\n");
            }
            else if(s2[0]=='A')
            {
                scanf("%d %d %d %s",&a,&b,&len,s3);
                uni(a,b,s[s3[0]-'A'],len);
                //printf("%d %s\n",m,s2);
            }
        }
    }
    return 0;
}