POJ 1182 解题报告

这道题似乎是union-find的经典题目。

这道题我在捕食关系上面弄得很乱，（改改说不定能弄对），最后看了discuss上面一个同学的解法写的，最后和这个同学的代码就一模一样了：http://poj.org/showmessage?message_id=344806

大致理解为：对任意一个编号为i的动物，它能捕食的动物集合和N+i都是一个集合的， 捕食它的动物集合和2N+i是一个集合的。这样总的集合大小为3N，编号为0~3N-1。好处是理解比较清晰。

判断两个动物(X, Y)可不可能是同一个类型的，只需要比较下Y和X捕食的动物或捕食X的动物在不在一个集合里面就可以了。如果可能是一个类型的，那就是真话，把X，Y放到一个集合里面，同时， X+N和Y+N，以及X+2N和Y+2N也要并起来。

判断两个动物(X,Y)是不是X能够捕食Y，也比较清晰。即看Y是不是和X或者捕食X的动物在一个集合里面。如果不能的话，那么X是可能会捕食Y的。这样，把X和捕食Y的放在一个集合里面。同理，把捕食X的和Y捕食的并起来，把X捕食的和Y并起来。

1182

Accepted

1796K

297MS

G++

2178B

/* 
ID: thestor1 
LANG: C++ 
TASK: poj1182 
*/
#include <iostream>
#include <fstream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <limits>
#include <string>
#include <vector>
#include <list>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <cassert>

using namespace std;

int findset(int X, std::vector<int> &parent)
{
	if (parent[X] != X)  
	{  
		parent[X] = findset(parent[X], parent);
	}  
	return parent[X];
}

void unionset(int X, int Y, std::vector<int> &parent, std::vector<int> &rank)
{
	int rx = findset(X, parent);
	int ry = findset(Y, parent);
	if (rx == ry)
	{
		return;
	}
	if (rank[rx] < rank[ry])
	{
		parent[rx] = ry;
	}
	else if (rank[ry] < rank[rx])
	{
		parent[ry] = rx;
	}
	else
	{
		parent[rx] = ry;
		rank[ry]++;
	}
}

int main()
{
	// std::ios::sync_with_stdio(false);
	int N, K;
	// cin >> N >> K;
	scanf("%d%d", &N, &K);
	
	std::vector<int> parent(3 * N);
	std::vector<int> rank(3 * N);
	for (int i = 0; i < 3 * N; ++i)
	{
		parent[i] = i;
		rank[i] = 0;
	}

	int nlies = 0;
	for (int i = 0; i < K; ++i)
	{
		int D, X, Y;
		// cin >> D >> X >> Y;
		scanf("%d%d%d", &D, &X, &Y);
		
		if (X < 1 || Y < 1 || X > N || Y > N)
		{
			nlies++;
			continue;
		}

		if (X == Y)
		{
			if (D == 2)
			{
				nlies++;
			}
			continue;
		}

		X--, Y--;

		// int rx = findset(X, parent);
		int ry = findset(Y, parent);
		// X and Y are the same group
		if (D == 1)
		{	
			if (findset(X + N, parent) == ry || findset(X + 2 * N, parent) == ry)
			{
				nlies++;
			}
			else
			{
				unionset(X, Y, parent, rank);
				unionset(X + N, Y + N, parent, rank);
				unionset(X + 2 * N, Y + 2 * N, parent, rank);
			}
		}
		else
		{
			// X eats Y
			assert (D == 2);
			if (findset(X, parent) == ry || findset(X + 2 * N, parent) == ry)
			{
				nlies++;
			}
			else
			{
				unionset(X, Y + 2 * N, parent, rank);
				unionset(X + N, Y, parent, rank);
				unionset(X + 2 * N, Y + N, parent, rank);
			}
		}
	}

	// cout << nlies << endl;
	printf("%d\n", nlies);
	return 0;  
}