ZOJ 3261 Connections in Galaxy War 【逆向並查集】

ZOJ 3261 Connections in Galaxy War

題目鏈接：vjudge傳送門

題目大意：
一共有N個星球，每個星球有一定的power值，兩個星球之間可以建立聯繫，當某個星球尋求幫助時，是向與它有直接或間接聯繫的星球中且power值最大的星球尋求幫助，如果兩個星球之間的聯繫被destory，則這兩個星球將失去直接聯繫，只可能存在間接聯繫。

具體思路：
完全想不到，還可以離線逆向操作看成並插件，太強了。。。
把destory邊，看成合並。初始化的時候，對給定的邊進行並集操作（除去將會被destory的邊），然後把問題逆向依次考慮，遇到destory的時候就把邊加入集合。最後把答案輸出即可。

注意：相鄰的case之間要輸出空行，路徑壓縮的時候，相同的power值時，編號小的作爲祖先。

思路來源：https://blog.csdn.net/include_peng/article/details/86571090

具體代碼：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
const int N = 1e4 + 10, M = 2e4 + 10, Q = 5e4 + 10;
int fa[N],power[N],ans[Q];
map<pair<int, int>, int> mp;	//離散存儲
int n, m, q;
struct Node {
	int flag;	//0表示query u,1表示destory u-v
	int u, v;
	void setNode(int flag, int u, int v) { this->flag = flag, this->u = u, this->v = v; }
}ques[Q];

struct Node2 {
	int flag;
	int u, v;
}edg[M];	//保存兩星球中的通道，flag表示存在否

int find(int x)
{
	if (fa[x] == -1)
		return x;
	return fa[x] = find(fa[x]);
}
void unite(int u, int v)
{
	int fu = find(u);
	int fv = find(v);
	if (fu == fv)return;
	if (power[fu] > power[fv] || power[fu] == power[fv] && fu<fv)
		fa[fv] = fu;
	else fa[fu] = fv;
}
void init()
{
	memset(fa, -1, sizeof(fa));
	for (int i = 0; i < m; i++)
	{
		if (edg[i].flag) {
			unite(edg[i].u, edg[i].v);
		}
	}
}
int main()
{
	int flag = 0;	//控制連續的測試數據間隔空行
	while (~scanf("%d", &n))
	{
		mp.clear();
		for (int i = 0; i < n; i++)scanf("%d", &power[i]);
		scanf("%d", &m);
		for (int i = 0; i < m; i++)
		{
			int u, v;
			scanf("%d%d", &u, &v);
			if (u > v)swap(u, v);
			mp[make_pair(u, v)] = i;
			edg[i].flag = 1;	//存在邊
			edg[i].u = u, edg[i].v = v;
		}
		int q;
		scanf("%d", &q);
		for (int i = 0; i < q; i++)
		{
			char str[10];
			scanf("%s", str);
			if (str[0] == 'q') {
				int s;
				scanf("%d", &s);
				ques[i].setNode(0, s, -1);
			}
			else {
				int u, v;
				scanf("%d%d", &u, &v);
				if (u > v)swap(u, v);
				ques[i].setNode(1, u, v);
				int tmp = mp[make_pair(ques[i].u, ques[i].v)];
				edg[tmp].flag = 0;	//逆過來看，該邊不存在
			}

		}
		init();	//初始化邊集
		for (int i = q-1; i >= 0; i--)
		{
			if (ques[i].flag) {	//destory
				unite(ques[i].u, ques[i].v);
			}
			else {
				int tmp = find(ques[i].u);
				if (power[tmp] > power[ques[i].u])
					ans[i] = tmp;
				else
					ans[i] = -1;
			}
		}
		if (flag) {
			printf("\n");
		}
		else {
			flag = 1;
		}
		for(int i=0;i<q;i++)
			if(ques[i].flag==0)printf("%d\n", ans[i]);
	}
	return 0;
}