POJ 1984 Navigation Nightmare（加权并查集）_navigation nightmare tarjan-优快云博客

题目链接：
POJ 1984 Navigation Nightmare
题意：
平面上有n个点，每个点可以在东西南北四个方向上与另外一个点连接，（每个点最多和四个直接连通）
然后有m条语句用以表示这n个点之间的位置关系：a,b,d,S表示a点在b点南方，距离为d（N,W,E分别表示北，西，东）
接着有k条语句来查询：a，b，t：由m条语句中的前t条能否得出a，b两点间的哈夫曼距离，
如果由m条语句中的前t条得出a，b两点不连通，则结果为-1.
分析：
先查询按照查询的t由小到大排序，然后将读入数据读入并查集到相应次序，两个点是否连通可通过find来判断。
x[i]，y[i]分别表示点i到根节点在横向和竖向的距离。
那么如果uu，vv两点连通距离就是abs(x[uu]-x[vv])+abs(y[uu]-y[vv]);
但是输出需要按照查询的顺序输出，所以可以把查询语句用结构体存储，并且结构体中一个变量用于存储查询顺序，
那么计算每条语句的ans时，把查询语句结构体数组按照t排序，输出时，再把查询语句按照查询顺序排序输出。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int maxn=40010;
const int maxk=10010;

int pre[maxn],x[maxn],y[maxn],n,m,k,a,b,d,tx,ty;
char ss[10];

struct Query{
    int index,u,v,t,ans;//ans是该条查询语句的答案
}query[maxk];

struct Read{
    int u,v,xx,yy;
}read[maxn];

bool cmp1(Query Q1,Query Q2)//按查询语句的t排序
{
    if(Q1.t==Q2.t) return Q1.index<Q2.index;
    return Q1.t<Q2.t;
}

bool cmp2(Query Q1,Query Q2)//按查询语句的查询顺序排序
{
    return Q1.index<Q2.index;
}

int find(int u)
{
    if(pre[u]==u) return u;
    int tmp=find(pre[u]);
    x[u]=x[u]+x[pre[u]];
    y[u]=y[u]+y[pre[u]];
    return pre[u]=tmp;
}

void mix(int a,int b,int tx,int ty)
{
    int fa=find(a);
    int fb=find(b);
    if(fa!=fb)
    {
        pre[fa]=fb;
        x[fa]=x[b]-x[a]+tx;
        y[fa]=y[b]-y[a]+ty;
    }
}

int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif
    while(~scanf("%d%d",&n,&m)&&n)
    {
        for(int i=0;i<=n;i++)
            pre[i]=i;
        memset(x,0,sizeof(x));
        memset(y,0,sizeof(y));
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d%s",&a,&b,&d,ss);
            if(ss[0]=='E'){tx=d;ty=0;}//这里对方向做了统一规定
            else if(ss[0]=='W'){tx=-d;ty=0;}
            else if(ss[0]=='S'){tx=0;ty=d;}
            else if(ss[0]=='N'){tx=0;ty=-d;}
            read[i].u=a;
            read[i].v=b;
            read[i].xx=tx;
            read[i].yy=ty;
        }
        scanf("%d",&k);
        for(int i=1;i<=k;i++)
        {
            scanf("%d%d%d",&query[i].u,&query[i].v,&query[i].t);
            query[i].ans=0;
            query[i].index=i;//查询顺序标记
        }
        sort(query+1,query+k+1,cmp1);
        int now=1;
        for(int i=1;i<=k;i++)
        {
            for(int j=now;j<=query[i].t;j++)
            {
                a=read[j].u;
                b=read[j].v;
                tx=read[j].xx;
                ty=read[j].yy;
                mix(a,b,tx,ty);
            }
            int uu=query[i].u;
            int vv=query[i].v;
            if(find(uu)==find(vv))//uu和vv连通
            {
                query[i].ans=abs(x[uu]-x[vv])+abs(y[uu]-y[vv]);
            }
            else query[i].ans=-1;
            //printf("uu=%d x[uu]=%d y[uu]=%d vv=%d x[vv]=%d y[vv]=%d \n",uu,x[uu],y[uu],vv,x[vv],y[vv]);
            //printf("************\n");
            now=query[i].t+1;
        }
        sort(query+1,query+1+k,cmp2);
        for(int i=1;i<=k;i++)
            printf("%d\n",query[i].ans);
    }
    return 0;
}