寒假算法每日一题2022.12.30

寒假算法每日一题2022.12.30

4818. 奶牛大学 - AcWing题库

Farmer John 计划为奶牛们新开办一所大学！

有 N 头奶牛可能会入学。

每头奶牛最多愿意支付 ci 的学费。

Farmer John 可以设定所有奶牛入学需要支付的学费。

如果这笔学费大于一头奶牛愿意支付的最高金额，那么这头奶牛就不会入学。

Farmer John 想赚尽可能多的钱，从而可以给他的讲师提供一笔可观的工资。

请求出他能赚到的钱的数量，以及此时应当收取多少学费。

输入格式
输入的第一行包含 N。

第二行包含 N 个整数 c1,c2,…,cN，其中 ci 是奶牛 i 愿意支付的最高学费金额。

输出格式
输出 Farmer John 可以赚到的最大金额以及最优情况下他应该收取的学费。如果有多个解，输出收取学费最小的解。

注意这个问题涉及到的整数可能需要使用 64 位整数型（例如，Java 中的 “long”，C/C++ 中的 “long long”）。

数据范围
$1\le N\le 105,$
$1\le ci\le 106。$

输入样例：

4
1 6 4 6

输出样例：

12 4

样例解释
如果 Farmer John 收费 4，那么 3 头奶牛将会入学，从而使他赚取 3×4=12 的金额。

解题思路：
我们先将奶牛愿意支付的最高金额从小到大排列，假设学费在任意两个相邻奶牛ai 和 ai+1之间(0<=i<n)，学费为x，奶牛的总数为n，则赚到的钱为 x * (n - i + 1)

这样分析的话，我们可以得到学费的可能是有无数多种的，因为学费可以是小数，但是学费无论定在 ai 和 ai+1之间（ai,ai+1）任意个点x，可以入学的奶牛都是 ai+1 之后的奶牛，数量是不变的。因此要得到总收益最多，必然有x == ai+1。

所以，学费的可能值只能在ai上，因此可能的学费有n中。

我们可以从小到大枚举奶牛，计算当前奶牛的最高支付金额作为学费的总收益，取最大值即为答案。

import java.util.Arrays;
import java.util.Scanner;

// AcWing4818 奶牛大学 （贪心）
public class Day1230 {
	static final int N = 100010;
	static int[] w = new int[N];
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		for(int i = 0; i < n ; i++) {
			w[i] = sc.nextInt();
		}
		Arrays.sort(w,0,n);
		long total = 0; // 总学费
		long free = 0; // 学费
		for(int i = 0; i < n;i++) {
			long tmp = (long)w[i] * (n - i);
			if(tmp > total) {
				total = tmp;
				free = w[i];
			}
		}
		
		System.out.println(total + " " + free);
		sc.close();
	}

}