Codeforces Round #634 (Div. 3) C. Two Teams Composing

题目


题意，给你一个n个元素，问，能否把这个n个元素分成俩个类别，一个类别中的数全部相等，另外一个全部不相等，且俩个类别元素总数相等(这俩个类别中可以有相等的)。问可以分成的每个类别的个数是最大多少。
思路;统计相同的最多元素个数和不同的个数，再做判断即可
AC代码

#include <bits/stdc++.h>
using namespace std;
#define NewNode (ListNode *)malloc(sizeof(ListNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 3e5 + 50;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const unsigned long long mod = 998244353;
const int II = 3.1415926535;
typedef long long ll;
typedef unsigned long long ull;
typedef pair <int,int> pii;
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t;
    cin >> t;
    while(t--)
    {
        int n;
        cin >> n;
        int arr[n+5] = {0},vis[n+5] = {0},Max = 0,MaxMax = 1;//Max种类数，MaxMax为某个相同的数的最多次数
        for(int i = 0;i < n;i++)
        {
            cin >> arr[i];
            if(vis[arr[i]])
                MaxMax = max(MaxMax,vis[arr[i]]+1);
            else
                Max++;
            vis[arr[i]]++;
        }
        MaxMax--,Max--;//因为MaxMax + Max肯定会多算了一个人，因为MaxMax的数也算一个种类数所以先减，后面再判断。
        if(MaxMax == Max || MaxMax+1 == Max)
            cout << Max << endl;
        else if(MaxMax == Max+1)
            cout << MaxMax << endl;
        else
            cout << min(MaxMax+1,Max+1) << endl;//还有一个人没有加上去。
    }
}