POJ 1984 Navigation Nightmare

並查集的應用，求解兩個點的曼哈頓距離

dx記錄子節點相對該集合代表的x偏移量，dy記錄子節點相對該集合代表的y偏移量

 

更新的時候注意相對偏移量的轉移即可，並且find操作時更新dx,dy

對於邊a->b，將find(a)的父節點指向b，那麼就可以在下次find操作時更新a原來集合中的點的dx,dy值

 

代碼：

#include<iostream> #include<memory.h> #include<string> #include<cstdio> #include<algorithm> #include<math.h> #include<stack> #include<queue> #include<vector> #include<map> using namespace std; const int MAX=40006; int fa[MAX],dx[MAX],dy[MAX],res[MAX],n,m; struct node { int u,v,w; char dir[4]; }g[MAX]; struct Node { int u,v,id,No; }Q[MAX]; int cmp(Node a,Node b) { return a.id<b.id; } int find(int x) { if(fa[x]!=x) { int t=fa[x]; fa[x]=find(fa[x]); dx[x]+=dx[t]; dy[x]+=dy[t]; } return fa[x]; } void solve() { int i,j=1,x,y; for(i=1;i<=m;i++) { while(j<=Q[i].id) { x=find(g[j].u); y=find(g[j].v); //if(x==y) //continue; if(g[j].dir[0]=='N') { dx[x]=-dx[g[j].u]; dy[x]=-dy[g[j].u]-g[j].w; fa[x]=g[j].v; } else if(g[j].dir[0]=='S') { dx[x]=-dx[g[j].u]; dy[x]=-dy[g[j].u]+g[j].w; fa[x]=g[j].v; } else if(g[j].dir[0]=='E') { dx[x]=-dx[g[j].u]-g[j].w; dy[x]=-dy[g[j].u]; fa[x]=g[j].v; } else { dx[x]=-dx[g[j].u]+g[j].w; dy[x]=-dy[g[j].u]; fa[x]=g[j].v; } j++; } x=find(Q[i].u); y=find(Q[i].v); if(x!=y) res[Q[i].No]=-1; else { res[Q[i].No]=abs(dx[Q[i].u]-dx[Q[i].v])+abs(dy[Q[i].u]-dy[Q[i].v]); } } } int main() { int i,j,k; scanf("%d%d",&n,&m); for(i=1;i<=n;i++) { fa[i]=i; dx[i]=dy[i]=0; } for(i=1;i<=m;i++) { scanf("%d%d%d%s",&g[i].u,&g[i].v,&g[i].w,&g[i].dir); } scanf("%d",&m); for(i=1;i<=m;i++) { scanf("%d%d%d",&Q[i].u,&Q[i].v,&Q[i].id); Q[i].No=i; } sort(Q,Q+m,cmp); solve(); for(i=1;i<=m;i++) printf("%d/n",res[i]); return 0; }