小x玩游戏

题目大意

今天,小 x 因为太无聊,就在玩游戏。

这个游戏有两个队伍,然后他们在游戏里面打来打去。

但小 x 遇到了难题。他不知道自己的队友是谁。他只知道总共有两个队伍,每队有 \(n\) 个人和很多组击杀情况。

他想问你,现在他能否知道两个队伍分别有谁。

你可以帮助小 x 吗?

解题思路

并查集水题。

对名字进行哈希,或者用 map

设 \(x+2*n\) 为 \(x\) 的敌人,

则 \(x\) 和 \(y\) 是敌人,即表示为 \(fa[find(a + 2 * n)] = find(b),fa[find(b + 2 * n)] = find(a)\),因为敌人的敌人是朋友嘛。

最后判断是否有矛盾即可。

考场判断写错,惨丢 \(30pts\),啊!

AC CODE

#include <bits/stdc++.h>
using namespace std;

#define int long long

const int _ = 200007;

const int base = 233;

const int mod = 1e9 + 7;

int T, n, m;

int fa[_];

int k, f[_];

map<long long, int> q;

int sm[_];

void init()
{
	for(int i = 1; i <= k; ++i) q[f[i]] = 0;
	k = 0;
	memset(sm, 0, sizeof sm);
	for(int i = 1; i < _; ++i) fa[i] = i;
}

int find(int x)
{
	return fa[x] == x ? x : fa[x] = find(fa[x]);
}

char c[15];

signed main()
{
//	freopen("game.in", "r", stdin);
//	freopen("game.out", "w", stdout);
	scanf("%lld", &T);
	while(T--)
	{
		int ans = 0;
		scanf("%lld%lld", &n, &m);
		init();
		for(int i = 1; i <= m; ++i)
		{
			scanf("%s", c);
			int a, b;
			int x = 0;
			for(int j = 0, len = strlen(c); j < len; ++j)
			{
				x = (x * base % mod + (int)(c[j] - 'a' + 1)) % mod;
			}
			if(!q[x])
			{
				q[x] = ++k;
				f[k] = x;
				a = k;
			}
			else
			{
				a = q[x];
			}
			scanf("%s", c);
			x = 0;
			for(int j = 0, len = strlen(c); j < len; ++j)
			{
				x = (x * base % mod + (int)(c[j] - 'a' + 1)) % mod;
			}
			if(!q[x])
			{
				q[x] = ++k;
				f[k] = x;
				b = k;
			}
			else
			{
				b = q[x];
			}
			fa[find(a + 2 * n)] = find(b);
			fa[find(b + 2 * n)] = find(a);

		}
		if(2 * n > k)
		{
			puts("NO");
			continue;
		}
		for(int i = 1; i <= n * 2; ++i)
			sm[find(i)]++;
		bool flag = 0;
		for(int i = 1; i <= n * 2; ++i)
		{
			if(sm[i] == n)
			{
				printf("YES\n");
				flag = 1;
				break;
			}
		}
		if(!flag) printf("NO\n");
	}
}
上一篇:POJ1611 The Suspects ——并查集


下一篇:splay