洛谷题解：P10351 [PA2024] Liderzy

题目大意

对于一个长度为 $n$ 的序列，对其中的数随便排列，并将其分成 $k$ 个部分，要求每个部分中的个数最多的数的个数必须超过该部分的数的总数的一半，求 $k$ 最小为多少。

思路讲解

设序列中值为 $i$ 的数的个数为 $a_i$，其中 $a$ 的长度为 $m$，我们可以将前 $k$ 大的 $a_i$ 对应的 $i$ 作为 $k$ 个部分中的个数最多的数，剩下的数作为其余的数。

那么如何求出 $k$ 呢？

已知一个部分中的个数最多的数的个数为 $a$，那么其余的数必须严格小于 $a$，也就是小于等于 $a-1$，那么 $k$ 个部分中每个部分中的个数最多的数的个数之和若为 $b$，那么其余的数的个数必须小于等于 $b-k$。

那么 $k$ 也很好求出来了。我们先把 $a$ 从大到小排序，让 $i$ 从 $1$ 开始从小到大遍历，第一个满足

$$\sum _{j=1}^i{a}_j-i\ge \sum _{j=i+1}^m{a}_j$$

的 $i$ 就是我们要求的 $k$。

AC 代码

#include<bits/stdc++.h>
using namespace std;
long long a[500001],b[500001]/*b 是前缀和数组，以快速求和*/;
bool cmp(long long x,long long y){
	return x>y;
}
int main(){
	long long n,m,x;
	cin>>n;
	for(long long i=1;i<=n;i++){
		scanf("%lld",&x);
		if(a[x]==0){
			m++;
		}
		a[x]++;
	}
	sort(a+1,a+1+500000,cmp);//从大到小排序 
	for(long long i=1;i<=m;i++){
		b[i]=b[i-1]+a[i];//求前缀和 
	}
	for(long long i=1;i<=m;i++){
		if(b[i]-i>=b[m]-b[i]/*相当于题解中的不等式*/){
			cout<<i;
			return 0;
		}
	}
	return 0;
}