【poj1984】 Navigation Nightmare

http://poj.org/problem?id=1984 (题目链接)

题意

　　给出一棵树，这棵树是以平面直角坐标系为基准建立的，也就是每个节点最多只有上下左右4条边。现在动态建树，同时询问两点间的曼哈顿距离

Solution

　　一开始没看懂题，当做图写了个SPFA。。后来发现是树于是删掉重新写了个DFS。。最后又发现要动态建树。。尼玛。。又重新写了个带全并查集。。

　　注意询问并不保证时间是升序的，要按照给定询问顺序输出。

代码

// poj1984
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#define LL long long
#define MOD 100000000
#define inf 2147483640
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;

const int maxn=100010;
struct edge {int u,v,w,t;}e[maxn];
struct data {int u,v,id,d;}t[maxn];
int n,m,q,X[maxn],Y[maxn],ans[maxn],fa[maxn];

bool cmp(data a,data b) {
	return a.id<b.id;
}
int find(int x) {
	if (x==fa[x]) return x;
	int f=find(fa[x]);
	X[x]+=X[fa[x]];
	Y[x]+=Y[fa[x]];
	return fa[x]=f;
}
int main() {
	scanf("%d%d",&n,&m);
	char ch;
	for (int x,u,v,w,i=1;i<=m;i++) {
		scanf("%d%d%d %c",&u,&v,&w,&ch);
		if (ch=='N') x=0;else if (ch=='S') x=1;else if (ch=='W') x=2;else x=3;
		e[i]=(edge){u,v,w,x};
	}
	scanf("%d",&q);
	for (int i=1;i<=q;i++) scanf("%d%d%d",&t[i].u,&t[i].v,&t[i].id),t[i].d=i;
	sort(t+1,t+1+q,cmp);
	for (int i=1;i<=n;i++) fa[i]=i;
	int pp=1;
	for (int u,v,w,ty,i=1;i<=m;i++) {
		u=e[i].u,v=e[i].v,w=e[i].w,ty=e[i].t;
		int r1=find(u),r2=find(v);
		fa[r1]=v;
		if (ty==0) X[r1]-=X[u],Y[r1]=e[i].w-Y[u];
		else if (ty==1) X[r1]-=X[u],Y[r1]=-e[i].w-Y[u];
		else if (ty==2) X[r1]=-e[i].w-X[u],Y[r1]-=Y[u];
		else X[r1]=e[i].w-X[u],Y[r1]-=Y[u];
		while (t[pp].id==i && pp<=q) {
			if (find(t[pp].u)!=find(t[pp].v)) ans[t[pp].d]=-1;
			else ans[t[pp].d]=abs(X[t[pp].u]-X[t[pp].v])+abs(Y[t[pp].u]-Y[t[pp].v]);
			pp++;
		}
		if (pp==q+1) break;
	}
	for (int i=1;i<=q;i++) printf("%d\n",ans[i]);
	return 0;
}