codeforces 709 div2 C. Basic Diplomacy

C. Basic Diplomacy

题意

t组数据，n天，玩m场游戏，每次的mi场游戏只能和题目给的朋友玩，如果和某个朋友玩的场数大于m/2，那么其他朋友就会很不开心，认为你偏心(手动滑稽)，你不想让你的朋友不开心，问能否挑选最佳策略使得人人开心，能则输出Yes，并且输出每场你选的一起玩的朋友，否则输出No

输入

2
4 6
1 1
2 1 2
3 1 2 3
4 1 2 3 4
2 2 3
1 3
2 2
1 1
1 1

输出

YES
1 2 1 1 2 3 
NO

思路

每次都挑选给出的第一个朋友，因为超过m/2的只能存在一个，如果符合就直接输出，不符合就再遍历一遍把超过m/2的那个人换一下，直到小于m/2或者结束
坑点:我试了半天才知道严格大于m/2是个什么东西
举例:x严格大于3/2，x = 3;不知道怎么算的，似乎是直接四舍五入的，也就是说ans > (x + 1) / 2

AC代码

#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
const int N = 2e5+10;
int a[N];
vector<int>v[100010];

int main()
{
	int t;
	cin >> t;
	while(t--)
	{
		map<int, int>mp;
		int n, m, x, cnt = 0, y;
		cin >> n >> m;
		for(int i=1; i<=m; i++)
		{
			cin >> x;
			v[i].clear();
			for(int j=1; j<=x; j++)
			{
				cin >> y;
				v[i].push_back(y);
			}
			a[i] = v[i][0];
		}
		int flag = 0, vis = 0;
		for(int i=1; i<=m; i++)
		{
			mp[a[i]]++;
			if(mp[a[i]] > (m + 1) / 2)
			{
				flag = mp[a[i]];
				vis = a[i];
			}
		}
		if(flag > (m + 1) / 2)
		{
			for(int i=1; i<=m; i++)
			{
				if(v[i].size() > 1 && v[i][0] == vis)
				{
					a[i] = v[i][1];
					flag--;
				}
				if(flag <= m / 2)
					break;
			}
		}
		if(flag > (m + 1) / 2)
			puts("NO");
		else
		{
			puts("YES");
			for(int i=1; i<=m; i++)
			{
				cout << a[i] << " ";
			}
			puts("");
		}			
	}
	return 0;
}