poj 1984 Navigation Nightmare 并查集

题意 每次给两点之间的方向和距离 求每次询问的曼哈顿距离

太久没做 想了很久觉得权值只能有正负两个值储存两个方向 其实用两个权值储存就好

有两个权值 所以在合并的时候一定要记得两个权值都要处理

#include <iostream>
#include <algorithm>
using namespace std;
///一个权值不能确定哪边 所以存两个
const int N=1e5;
int father[N],rx[N],ry[N],n;
int Find(int x)
{
    int tx=father[x];
    if(x==tx)return x;
    father[x]=Find(tx);
    rx[x]+=rx[tx];
    ry[x]+=ry[tx];
    return father[x];
}
void unite(int x,int y,int d,int flag)
{
    int tx=Find(x);
    int ty=Find(y);
    if(tx==ty)return;
    father[ty]=tx;
    if(flag==1)
    {
        rx[ty]=-rx[y]-d+rx[x];
        ry[ty]=-ry[y]+ry[x];///3rd
    }
    else if(flag==2)
    {
        rx[ty]=-rx[y]+d+rx[x];
        ry[ty]=-ry[y]+ry[x];
    }
    else if(flag==3)
    {
        ry[ty]=-ry[y]-d+ry[x];
        rx[ty]=-rx[y]+rx[x];
    }
    else if(flag==4)
    {
        ry[ty]=-ry[y]+d+ry[x];
        rx[ty]=-rx[y]+rx[x];
    }
}
void init()
{
    for(int i=0;i<=n;i++)
    {
        father[i]=i;
        rx[i]=0;ry[i]=0;
    }
}
struct input1{
    int f1,f2,di;
    char fx;
}a[N];
struct input2{
    int f1,f2,i,id,ans;
    bool operator<(const input2&b)const{
        return i<b.i;
    }
}b[N];
bool cmp(input2 a,input2 b)
{
    return a.id<b.id;
}
int main()
{
    ios::sync_with_stdio(false);
    int m,k;
    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++)cin>>a[i].f1>>a[i].f2>>a[i].di>>a[i].fx;
        cin>>k;
        for(int i=0;i<k;i++)
        {
            cin>>b[i].f1>>b[i].f2>>b[i].i;
            b[i].id=i;
        }
        sort(b,b+k);
        int nn=0;
        for(int i=0;i<m;i++)
        {
            if(nn==k)break;
            if(a[i].fx=='E')
            {
                unite(a[i].f1,a[i].f2,a[i].di,1);
            }
            else if(a[i].fx=='W')
            {
                unite(a[i].f1,a[i].f2,a[i].di,2);
            }
            else if(a[i].fx=='N')
            {
                unite(a[i].f1,a[i].f2,a[i].di,3);
            }
            else if(a[i].fx=='S')
            {
                unite(a[i].f1,a[i].f2,a[i].di,4);
            }
            while(i+1==b[nn].i)///2nd
            {
                int tx=b[nn].f1;
                int ty=b[nn].f2;
                if(Find(tx)!=Find(ty))
                    b[nn].ans=-1;
                else
                {
                    int ans=abs(rx[tx]-rx[ty])+abs(ry[tx]-ry[ty]);
                    b[nn].ans=ans;
                }
                nn++;
            }
        }
        sort(b,b+k,cmp);///1st
        for(int i=0;i<k;i++)
            cout<<b[i].ans<<endl;
    }
    return 0;
}
/*
6 3
4 1 1 N
6 3 1 E
6 1 1 W
1
1 3 3
//2
4 3
1 2 1 E
3 4 2 S
2 4 1 N
2
1 3 3
1 4 3
//4 2
*/