题目链接:1984 -- Navigation Nightmare
中文题意:
有n个网格状的农田,每个农田之间有距离,会依次给出关系,在给出关系后询问两个农田之间的曼哈顿距离是多少?(点(x1,y1)和点(x2,y2)的哈密顿距离是|x1-x2|+|y1-y2|)
若无法判断则输出-1。
这道题目是用带权并查集来解决的,两个点的哈密顿距离能否计算取决于他们是否具有相同的参考系,也就是说他们是否在一个集合中,我们可以通过集合中每个点与该集合代表元素的关系来确定集合内任意两点的相对位置。由于这是一个二维关系,我们可以将其拆分为两个一维关系来看,最后合并即为结果。
比如一个集合包含a,b,c……,集合代表元素是c,其中a在c西边3,下边5,而b在c东边1,上边1,则我们就能够知道a和b的哈密顿距离为4+6=10,而这道题就是利用这样的思想来求解的,下面我主要来说一下如何进行并查集合并
x和y位于两个不同的集合,现在要令x所在集合的代表元素作为新的集合的代表元素,现在知道x在其代表元素东边dx[x],而y在其代表元素东边dx[y],且y在x东边wx,则由图易知dx[f2]=dx[x]+wx-dx[y]同理可得南北方向的合并方式,还需要注意的一点是题目中没有说明询问时间是从小到大的,所以需要加个排序,题目分析到这基本上就分析完了,下面看代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
const int N=40003;
int dx[N],dy[N],fu[N],n,m;
int a[N],b[N],l[N];
char op[N][5];
struct node{
int u,v,t,id,ans;
}p[N];
bool cmp1(node a,node b)
{
return a.t<b.t;
}
bool cmp2(node a,node b)
{
return a.id<b.id;
}
void init()
{
for(int i=1;i<=n;i++)
fu[i]=i;
memset(dx,0,sizeof dx);
memset(dy,0,sizeof dy);
}
int find(int x)
{
int t=fu[x];
if(x==fu[x]) return x;
fu[x]=find(fu[x]);
dx[x]+=dx[t];
dy[x]+=dy[t];
return fu[x];
}
void merge(int x,int y,int wx,int wy)
{
int f1=find(x),f2=find(y);
fu[f2]=f1;
dx[f2]=dx[x]+wx-dx[y];
dy[f2]=dy[x]+wy-dy[y];
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
init();
for(int i=1;i<=m;i++)
scanf("%d%d%d%s",&a[i],&b[i],&l[i],op[i]);
int k;
scanf("%d",&k);
for(int i=1;i<=k;i++)
scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].t),p[i].id=i;
sort(p+1,p+k+1,cmp1);
int cnt=1;
for(int i=1;i<=m;i++)
{
if(op[i][0]=='E') merge(a[i],b[i],l[i],0);
else if(op[i][0]=='W') merge(a[i],b[i],-l[i],0);
else if(op[i][0]=='N') merge(a[i],b[i],0,l[i]);
else merge(a[i],b[i],0,-l[i]);
while(p[cnt].t<=i&&cnt<=k)
{
int f1=find(p[cnt].u),f2=find(p[cnt].v);
if(f1!=f2) p[cnt].ans=-1;
else p[cnt].ans=abs(dx[p[cnt].u]-dx[p[cnt].v])+abs(dy[p[cnt].u]-dy[p[cnt].v]);
cnt++;
}
}
sort(p+1,p+k+1,cmp2);
for(int i=1;i<=k;i++)
printf("%d\n",p[i].ans);
}
return 0;
}