POJ 1984 Navigation Nightmare【帶權並查集 二維】

POJ 1984 Navigation Nightmare

題目鏈接： vjudge傳送門

題目大意：
依次給定二維座標軸上的若干點的相對座標，問在指定的前i次數據前，能否得到兩點之間的距離(橫縱座標之差的絕對值之和)。

具體思路：
並查集，用兩個權值數組hori[],vert[]，分別維護水平方向和垂直方向上相對於根節點的相對距離，然後對每問兩點的距離進行建樹和輸出。如果是一次query就建圖一次，肯定是不行的。我第一次就是這麼做的，會超時。
因此需要先把需要求的每對點，保存下來，然後依據需要的數據從小到大進行排序，就可以通過一次建圖，然後在指定的數據量時求解並保存答案。最終依次輸出答案。

需要注意的是，輸出的答案順序必須與題目詢問的順序相同。
感覺自己說得很亂，應該只有自己能看得懂，還是看代碼吧

具體代碼：

#include <cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
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;
}