https://codeforces.com/problemset/problem/1335/C
题意:给定 n 个学生的技能,将其中的学生分为两组,两组的学生人数应该相等,且第一组的学生技能值应互不相同,第二组的学生技能值应该都相同,求最大的学生人数
思路:显然应该用二分查找学生人数,且二分的右端点应该小于 n/2,那么 check 函数应该如何写呢?我们用 map 保存每种技能的学生数,然后我们首先遍历整个 map ,找值不等于 x (要check的每组学生人数) 的人数一共有多少,且当 mp[i] > x 时,mp[i] --,因为当这个学生人数大于 x 时,它在被选入第一组之后可能剩下的人数会等于 x ,也就是可以被选入第二组,此时我们得到了 cnt,也就是可以入选第一组的人一共有多少,然后接下来我们就要分两种情况讨论了
第一种情况:cnt >= x,也就是第一组的人数可以满足,此时我们考虑第二组就简单了,只需要判 断 map 中是否还有人数大于等于 x 的,如果有,直接返回 true
第二种情况:cnt < x,也就是第一组人数还不满足,此时我们要判断 map 中人数大于等于 x 的一 共有多少种,如果 cnt1 > ( x-cnt ),也就是在大于等于 x 的中抽调人去第一组之后,还能够有至少一组来满足第二组
时间复杂度为 O( n * logn )
以下是代码实现:
#include<iostream>
#include<map>
#include<algorithm>
using namespace std;
const int N=2e5+10;
map<int,int>mp1,mp2;
int num[N];
bool check(int x)
{
mp2=mp1;
int cnt=0,cnt1=0;
for(auto i:mp2)
{
if(1<=i.second&&i.second<=x-1 || i.second>x)
{
cnt++;
if(i.second>x)
{
mp2[i.first]--;
}
}
}
if(cnt>=x)//第一组人数满足
{
for(auto i:mp2)
{
if(i.second>=x)//若第二组条件也能满足
{
return true;
}
}
}
else
{
for(auto i:mp2)
{
if(i.second>=x)
{
cnt1++;
}
}
if(cnt1>(x-cnt))//在抽调人之后仍有一组能满足第二组的条件
{
return true;
}
}
return false;
}
void solved()
{
mp1.clear(),mp2.clear();
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>num[i];
mp1[num[i]]++;
}
if(n==1)
{
cout<<0<<endl;
return ;
}
int l=1,r=n/2,mid=0,ans=0;
while(l<=r)
{
mid=(l+r)/2;
if(check(mid))
{
ans=mid;
l=mid+1;//看看有没有更大的
}
else
{
r=mid-1;
}
}
cout<<ans<<endl;
return ;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int t;
cin>>t;
while(t--)
{
solved();
}
return 0;
}