【集合论】B. Groups

题目来源

Problem - 1598B - Codeforceshttps://codeforces.com/problemset/problem/1598/B

题干

 

官方解释

注释

这道题看到数据量就应该想到是一个暴力算法

用两个指针i,j来迭代一个固定的日期a,b，最多会迭代10次

然后开始给选这两个日期的学生进行分类

四个集合：

1.a,b都不选的

2.只选a的

3.只选b的

4.a,b都选的

第一个集合非空的话，就代表目前迭代的日期a,b不符合

然后记录2，3集合的元素数分别为cnt1和cnt2

如果cnt1或cnt2超过n/2的话，就代表上另一个日期课的学生少于n/2了，则不符合

如果cnt1&&cnt2 <n/2，则代表集合4非空，可以从集合4分配学生分别到cnt1和cnt2,满足cnt1=cnt2

代码段

void solve()//group
{
	int n,cnta = 0, cntb = 0;
	cin >> n;
	vector<vector<int>> a(n, vector<int>(5));
	for (int i = 0; i < n; i++)
		for (int j = 0; j < 5; j++)
			cin >> a[i][j];
	for (int i = 0; i < 4; i++)
	{
		for (int j = i+1; j < 5; j++)
		{
			for (int k = 0; k < n; k++)
			{
				if (a[k][i] == 0 && a[k][j] == 0)goto re;
				if (a[k][i] == 1 && a[k][j] == 0)cnta++;
				if (a[k][i] == 0 && a[k][j] == 1)cntb++;
				if (cntb > n / 2 || cnta > n / 2)goto re;
			}
			cout << "YES" << endl; return;
		re:
			cnta = 0, cntb = 0;
		}
	}
	cout << "NO" << endl;
}