【POJ 1984】Navigation Nightmare (并查集)

并查集的题，说是图论也算是吧。

题目中较易出错的两个地方：

一、index不一定是升序的，要注意按原题中的顺序输出结果。

比如：

8 7
1 2 1 S
3 4 5 S
5 6 8 S
7 8 2 S
3 2 3 N
7 6 13 N
5 4 7 N
6
1 3 1
1 4 5
3 4 2
4 5 7
1 8 6
1 8 7

应该输出的是

-1
9
5
7
-1
39

 

二、注意查询时可能一个时间查询多次。（如果出现这个错误的话，上面的那组数据中的最后一组估计不会得到正确的结果）

 

我写的时候进行了路径压缩，没写rank优化。

#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<climits> using namespace std; const int MAX=40010,MAX2=10100; int us[MAX],vs[MAX],ls[MAX]; char dirs[MAX]; struct Query { Query(){} Query(int a,int b,int t,int old):a(a),b(b),t(t),old(old){} int a,b,t,old,val; }; Query qs[MAX2]; template<int Max> class UnionFindSet { public: UnionFindSet() { memset(father,-1,sizeof(father)); } void Union(int u,int v,int disx,int disy) { int disx1,disx2,disy1,disy2; int fu=Find(u,disx1,disy1),fv=Find(v,disx2,disy2); ydis[fv]=disy1+disy-disy2;xdis[fv]=disx1+disx-disx2; father[fv]=fu; } int Find(int u,int & disx,int &disy) { if(father[u]==-1) { disx=0; disy=0; return u; } int dx,dy; int v=Find(father[u],dx,dy); xdis[u]=disx=xdis[u]+dx; ydis[u]=disy=ydis[u]+dy; father[u]=v; return v; } private: int father[Max]; int ydis[Max],xdis[Max]; }; bool sortbyt(const Query& q1,const Query& q2) { return q1.t<q2.t; } bool sortbyold(const Query& q1,const Query& q2) { return q1.old<q2.old; } UnionFindSet<MAX> u; void getDis(char dir,int len,int &disx,int &disy) { switch(dir) { case 'E':disx=len;disy=0;break; case 'W':disx=-len;disy=0;break; case 'N':disx=0;disy=-len;break; case 'S':disx=0;disy=len;break; } } int front=0; inline int myabs(int a) { return a>=0?a:-a; } int main() { //freopen("in.txt","r",stdin); int v,e,q; char dir; scanf("%d%d",&v,&e); for(int i=0;i!=e;i++) { scanf("%d%d%d %c",&us[i],&vs[i],&ls[i],&dirs[i]); } scanf("%d",&q); for(int i=0;i!=q;i++) { scanf("%d%d%d",&qs[i].a,&qs[i].b,&qs[i].t); qs[i].old=i; } sort(qs,qs+q,sortbyt); int dx,dy,dx2,dy2; for(int i=0;i!=e;i++) { getDis(dirs[i],ls[i],dx,dy); u.Union(us[i],vs[i],dx,dy); while(qs[front].t==i+1) { int fu=u.Find(qs[front].a,dx,dy),fv=u.Find(qs[front].b,dx2,dy2); if(fu!=fv) qs[front].val=-1; else qs[front].val=myabs(dx-dx2)+myabs(dy-dy2); ++front; } } sort(qs,qs+q,sortbyold); for(int i=0;i!=q;i++) printf("%d/n",qs[i].val); }