Candy Machine
题目链接
题意 :
给定一系类数组,求所有子集中的一个子集中满足严格大于这个子集的平均数的元素个数的最大值
思路一:
假设最终选择的集合的平均数不超过 k。
为使平均数不超过 k,应将 ≤ k 的数全部选入,然后贪心选
择 > k 的部分中最小的若干个数。
因此将 N 个数从小到大排序后,最优解一定是一个前缀。
可以先将这个数组从小到大排序,然后依次比较每次去掉最大的一个元素之后的结果,符合条件所得的ans的最大值为所求。
因为如果每次去掉的是最小的数组元素的话,所得数组的平均值会变大,最后该数组中能满足严格大于这个数组的平均值的元素个数就会变小;反之,平均值会变小,结果就会有变大的可能性。所以每次比较要去掉最大的数。
AC代码 :
#include <iostream>
#include <map>
#include <cstring>
#include <algorithm>
#include <math.h>
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;
ll a[N];
ll n, sum, ans;
int main()
{
IOS;
cin >> n;
for (ll i = 0; i < n; i++)
{
cin >> a[i];
sum += a[i];
}
sort(a, a + n);
double aver = sum * 1.0 / n;
for (ll i = 0; i < n; i++)
{
if ((double)a[i] > aver)
ans++;
}
for (ll i = n - 1; i >= 1; i--)
{
sum -= a[i];
aver = sum * 1.0 / (i)*1.0;
ll x = upper_bound(a, a + i - 1, aver) - a;
ans = max(ans, i - x);
}
cout << ans << endl;
return 0;
}
思路二
排序之后直接枚举每一个前缀,然后取最大值即可
void solve() {
int n; cin >> n;
vi a(n);
for (int i = 0; i < n; i++) cin >> a[i];
sort(all(a));
int ans = 0;
double pre = 0; // 前缀和
for (int i = 0; i < n; i++) { // 枚举每个前缀
pre += a[i];
double avg = pre / (i + 1);
int l = 0, r = i;
while (l < r) {
int mid = l + r + 1 >> 1;
if (cmp(a[mid], avg) <= 0) l = mid; // a[mid]≤avg
else r = mid - 1;
}
ans = max(ans, i - r);
}
cout << ans;
}
int main() {
solve();
}