题意
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;
}