UVA_793_Network Connections

#include<iostream>  
#include<sstream>  
#include<string>  
#include<vector>  
#include<list>  
#include<set>  
#include<map>  
#include<stack>  
#include<queue>  
#include<algorithm>  
#pragma warning(disable:4996)  
using std::cin;
using std::cout;
using std::endl;
using std::stringstream;
using std::string;
using std::vector;
using std::list;
using std::pair;
using std::set;
using std::multiset;
using std::map;
using std::multimap;
using std::stack;
using std::queue;
class DisjoinSet
{
private:
	vector<int>represent, rank;
public:
	DisjoinSet(const int &n) :represent(n), rank(n)
	{
		for (int i = 0; i < n; i++)
		{
			represent[i] = i;
		}
	}
	int find(int i)
	{
		return represent[i] = represent[i] == i ? i : find(represent[i]);
	}
	void merge(const int &x, const int &y)
	{
		auto a = find(x); auto b = find(y);
		if (rank[a] < rank[b])
		{
			represent[a] = b;
		}
		else
		{
			represent[b] = a;
			if (rank[b] == rank[a])
			{
				rank[a]++;
			}
		}
	}
	int count()
	{
		set<int>ret;
		for (size_t i = 0; i < represent.size(); i++)
		{
			ret.insert(find(i));
		}
		return ret.size();
	}
};
int main()
{
	//freopen("input.txt", "r", stdin);
	//freopen("output.txt", "w", stdout);
	int T; cin >> T;
	while (T--)
	{
		int n; cin >> n; cin.get();
		DisjoinSet union_set(n);
		string str;
		int yes = 0, no = 0;
		while (getline(cin, str))
		{
			if (str.empty())
			{
				break;
			}
			stringstream stream; stream << str;
			char order; stream >> order;
			if (order == 'c')
			{
				int first, second; stream >> first >> second;
				union_set.merge(first - 1, second - 1);
			}
			else
			{
				int first, second; stream >> first >> second;
				if (union_set.find(first - 1) == union_set.find(second - 1))
				{
					yes++;
				}
				else
				{
					no++;
				}
			}
		}
		cout << yes << ',' << no << endl;
		if (T)
		{
			cout << endl;
		}
	}
	return 0;
}

你可能感兴趣的:(uva,union_find)