POJ 1984 Navigation Nightmare【带权并查集 二维】

POJ 1984 Navigation Nightmare

题目链接： vjudge传送门

题目大意：
依次给定二维坐标轴上的若干点的相对坐标，问在指定的前i次数据前，能否得到两点之间的距离(横纵坐标之差的绝对值之和)。

具体思路：
并查集，用两个权值数组hori[],vert[]，分别维护水平方向和垂直方向上相对于根节点的相对距离，然后对每问两点的距离进行建树和输出。如果是一次query就建图一次，肯定是不行的。我第一次就是这么做的，会超时。
因此需要先把需要求的每对点，保存下来，然后依据需要的数据从小到大进行排序，就可以通过一次建图，然后在指定的数据量时求解并保存答案。最终依次输出答案。

需要注意的是，输出的答案顺序必须与题目询问的顺序相同。
感觉自己说得很乱，应该只有自己能看得懂，还是看代码吧

具体代码：

#include 
#include
#include
#include
using namespace std;
const int N = 4e4 + 10, M = 4e4;
struct Node {
     
	int  u, v, w, d;	//EWNS分别用1，2，3，4表示
	void setNode(int u, int v, int w, char c) {
     
		this->u = u, this->v = v, this->w = w;
		if (c == 'E')d = 1;
		else if (c == 'W')d = 2;
		else if (c == 'N')d = 3;
		else d = 4;
	}
}maps[N];
struct Node2{
     
	int u, v, lim, ind;
	void setNode2(int u, int v, int ind,int lim) {
      this->u = u, this->v = v, this->lim = lim, this->ind = ind; }
}test[M];
int fa[N], hori[N], vert[N], ans[M];
int n, m;
int find(int x)
{
     
	if (fa[x] == -1)
		return x;
	int t = find(fa[x]);
	hori[x] = hori[x] + hori[fa[x]];
	vert[x] = vert[x] + vert[fa[x]];
	return fa[x] = t;
}
void unite(int u, int v,int index)
{
     
	int fu = find(u);
	int fv = find(v);
	if (fu == fv)return;
	fa[fv] = fu;
	int d = maps[index].d;
	int w = maps[index].w;
	switch (d)
	{
     
		case 1: {
     
			hori[fv] = hori[u] + w - hori[v];
			vert[fv] = vert[u] - vert[v];
		} break;
		case 2: {
     
			hori[fv] = hori[u] - w - hori[v];
			vert[fv] = vert[u] - vert[v];
		} break;
		case 3: {
     
			hori[fv] = hori[u] - hori[v];
			vert[fv] = vert[u] + w - vert[v];
		} break;
		case 4: {
     
			hori[fv] = hori[u] - hori[v];
			vert[fv] = vert[u] - w - vert[v];
		} break;
	}
}
void init()
{
     
	memset(fa, -1, sizeof(fa));
	memset(hori, 0, sizeof(hori));
	memset(vert, 0, sizeof(vert));
}
bool cmp(Node2 a, Node2 b) {
      return a.lim < b.lim; }
int getAns(int u, int v)
{
     
	if (find(u) != find(v))return -1;
	return abs(hori[u] - hori[v]) + abs(vert[u] - vert[v]);
}
void makeSet()
{
     
	int cnt = 0;
	for (int i = 1; i <= m; i++)
	{
     
		if (find(maps[i].u) == find(maps[i].v))continue;
		unite(maps[i].u, maps[i].v, i);
		while (i == test[cnt].lim) {
     	//把while改成if会WR,因为可能出现同一个lim，不同的test询问情况
				ans[test[cnt].ind] = getAns(test[cnt].u, test[cnt].v);
				cnt++;
		}
	}
}

int main()
{
     
	while (~scanf("%d%d", &n, &m))
	{
     
		for (int i = 1; i <= m; i++)
		{
     
			int u, v, w;
			char c;
			scanf("%d%d%d %c", &u, &v, &w, &c);
			maps[i].setNode(u, v, w, c);
		}
		int num;
		scanf("%d", &num);
		for (int i=0;i<num;i++)
		{
     
			int u, v, limit;
			scanf("%d%d%d", &u, &v, &limit);
			test[i].setNode2(u, v, i,limit);
		}
		sort(test, test + num, cmp);
		init();
		makeSet();
		for(int i=0;i<num;i++)
			printf("%d\n", ans[i]);
	}
	return 0;
}