一道很水的UFS题,但是维护节点信息的时候却遇到了一个我以前从来没有想过也从来没有遇到过的问题:
以前在合并两个节点的时候总是先判断它们是否在同一集合中,然后再合并;这样在判断的过程中就更新了节点信息。但这道题有一个奇怪的地方时它不需要判断,于是我就也傻乎乎的直接合并了信息,导致合并的信息是滞后的。从而导致WA掉了,而且一直找不到原因,直到下了数据才搞明白。
这道题给了我两点启示:
①(直接的)在UFS题目中在Union操作之前一定要先更新节点信息。
②(深层次的)在写代码时一定要谨慎小心,精神集中,弄清各个变量的意义及其变化,才能避免出错。
代码:
#include<iostream>
using namespace std;
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
char * ptr=new char [1000001];
struct S{
int f1,f2,i,x;
}q[10001];
inline bool cmp(S a,S b){
return a.i<b.i;
}
int fa[40001],f1[40001],f2[40001],lth[40001],v[40001],h[40001],ans[10001];
inline int find(int x){
if(fa[x]!=fa[fa[x]]){
int ftr=fa[x];
fa[x]=find(fa[x]);
v[x]+=v[ftr];
h[x]+=h[ftr];
}
return fa[x];
}
inline void in(int &x){
while(*ptr<'0'||*ptr>'9')++ptr;
x=0;
while(*ptr>47&&*ptr<58)x=x*10+*ptr++-'0';
}
inline void in(char &x){
while(*ptr<'A'||*ptr>'Z')++ptr;
x=*ptr++;
}
char c[40001];
int main(){
//freopen("POJ1984.in","r",stdin);
int j=0,k,i=0,m,n;
ptr[fread(ptr,1,1000000,stdin)]='z';
in(n);
in(m);
++m;
while(++i<m){
in(f1[i]);
in(f2[i]);
in(lth[i]);
in(c[i]);
}
in(k);
for(i=0;i<k;++i){
in(q[i].f1);
in(q[i].f2);
in(q[i].i);
q[i].x=i;
}
sort(q,q+k,cmp);
q[k].i=-1;
for(i=0,++n;i<n;++i)fa[i]=i;
for(i=1;i<m;++i){
find(f2[i]);
switch(c[i]){
case 'E':
h[find(f1[i])]=lth[i]-h[f1[i]]+h[f2[i]];
v[fa[f1[i]]]=-v[f1[i]]+v[f2[i]];
break;
case 'W':
h[find(f1[i])]=-lth[i]-h[f1[i]]+h[f2[i]];
v[fa[f1[i]]]=-v[f1[i]]+v[f2[i]];
break;
case 'N':
v[find(f1[i])]=lth[i]-v[f1[i]]+v[f2[i]];
h[fa[f1[i]]]=-h[f1[i]]+h[f2[i]];
break;
case 'S':
v[find(f1[i])]=-lth[i]-v[f1[i]]+v[f2[i]];
h[fa[f1[i]]]=-h[f1[i]]+h[f2[i]];
break;
}
fa[fa[f1[i]]]=fa[f2[i]];
while(q[j].i==i){
if(find(q[j].f1)!=find(q[j].f2))ans[q[j].x]=-1;
else ans[q[j].x]=abs(v[q[j].f1]-v[q[j].f2])+abs(h[q[j].f1]-h[q[j].f2]);
++j;
}
}
for(i=0;i<k;++i)
printf("%d\n",ans[i]);
return 0;
}