以下是2022年-2023年USACO赛季第一个月的青铜组第一题,可以使用“计数数组+打擂台找最值”的思想,需要考生掌握一点算法才能通关。文章引用了官网题面,提供思考思路和代码,代码对每一句话做出解释,有相关疑问欢迎交流,祝大家都能通关USACO!
Farmer John 计划为奶牛们新开办一所大学!
有 N(1≤N≤10^5)头奶牛可能会入学。每头奶牛最多愿意支付 ci 的学费(1≤ci≤10^6)。 Farmer John 可以设定所有奶牛入学需要支付的学费。如果这笔学费大于一头奶牛愿意支付的最高金额,那么这头奶牛就不会入学。Farmer John 想赚尽可能多的钱,从而可以给他的讲师提供一笔可观的工资。请求出他能赚到的钱的数量,以及此时应当收取多少学费。
输入格式(从终端 / 标准输入读入):
输入的第一行包含 NN。第二行包含 NN 个整数 c1,c2,…,cN,其中 ci 是奶牛 i 愿意支付的最高学费金额。
输出格式(输出至终端 / 标准输出):
输出 Farmer John 可以赚到的最大金额以及最优情况下他应该收取的学费。如果有多个解,输出收取学费最小的解。
注意这个问题涉及到的整数可能需要使用 64 位整数型(例如,Java 中的 "long",C/C++ 中的 "long long")。
输入样例:
4 1 6 4 6输出样例:
12 4如果 Farmer John 收费 4,那么 3 头奶牛将会入学,从而使他赚取 3⋅4=12 的金额。
测试点性质:
- 测试点 2-4 满足 ci≤1,000。
- 测试点 5-8 满足 N≤5,000。
- 测试点 9-12 没有额外限制。
供题:Freddie Tang
思路:总学费最大值=单个人的学费*交的起的总人数
可以枚举出所有的组合,但是介于人数超过10^5,学费可以考虑遍历,那么这里就需要思考以下如何不嵌套循环、直接计算出人数,所以前缀和就登场了。优化的过程初期可以先从最简单的暴力枚举开始,毕竟暴力也可以拿8个测试点呢!
#include <cstdio>
#include <iostream>
using namespace std;
long long sum[1000500], a[1000500]; //前缀和 计数数组
int main(){
long long n, maxn = 0;
scanf("%lld", &n);
for(long long i = 1; i <= n; i++){
long long tmp;
scanf("%lld", &tmp);
a[tmp]++; //统计当前可以支付tmp学费的人数
if (maxn < tmp)maxn = tmp; //找到可支付的学费最大值,单纯为优化提速,也可用1e6
}
sum[maxn] = a[maxn]; //这里是一个小坑点,注意初始化赋值
for(long long i = maxn; i >= 1; i--){ //前缀和数组的倒置应用
sum[i - 1] = sum[i] + a[i - 1]; //学费>=i-1的人有sum[i-1]个
}
// 中单提一下:以下注释掉的这一段是为了调代码用的,帮助理解中间与核查中间计算过程
// cout << "maxn = "<< maxn << "\n";
// for(long long i = maxn; i >= 1; i--){
// cout << sum[i] << " "; //学费>=i-1的人有sum[i-1]个
// }
// cout << "\n";
//题目是为了找到总的学费最大值,那就可以匹配所有组合打个擂台记录下答案就好啦
long long ans = 0, fee = 0;
for(long long i = maxn; i >= 1; i--){
if (i * sum[i] >= ans){
ans = i * sum[i];
fee = i;
}
}
//过程中务必注意数据类型
printf("%lld %lld", ans, fee);
return 0;
}