POJ 1984 Navigation Nightmare（加权并查集）

题目链接：

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;
}