问题描述
输入输出
解题思路
根据题意知,当前项目的花费为max(kx+b,0)*x。由于k为负值,因此在一定范围内,抛物线必定有最高点。
那么我们只需要找到抛物线停止递增时的人数x即可,当项目每多一个人时,收益为(k(x+1)+b)(x+1)-(kx+b)x = k(2x+1)+b。
我们可以把每一个项目看做一个个篮子,把人一个个地往里面丢。
由于每个项目的k、b不同,因此往里面丢人时的收益也不同。
那么我们维护一个从大到小排序的优先队列,加入每个项目的(k、x、b),由k(2x+1)+b计算出每一个项目丢人进去的收益并排序,丢人时直接取出队列队头的项目即可。然后再将(k、x+1、b)入队。若队头元素<=0,即收益最高的项目收益<=0,说明已经达到了最大收益。
收益是可能超出int最大范围的,所以用long。
AC代码
import java.util.PriorityQueue;
import java.util.Scanner;
// 相当于项目的结构体
class Project implements Comparable<Project> {
// k、b为参数,x为项目当前人数
long k, x, b;
public Project(long k, long x, long b) {
this.k = k;
this.x = x;
this.b = b;
}
// 计算添加一个人的收益k(2x+1)+b
public long earn() {
return (2 * x + 1) * k + b;
}
// 实现Comparable接口的compareTo方法,用于比较项目收益
@Override
public int compareTo(Project p) {
return Long.compare(p.earn(), this.earn());
}
}
public class Main {
public static void main(String[] args) {
PriorityQueue<Project> queue = new PriorityQueue<>(); // 创建优先队列,结合compareTo使得从大到小排序
long cost = 0; // 记录项目总收益(最大花费)
Scanner scan = new Scanner(System.in);
int n = scan.nextInt();
int m = scan.nextInt();
// 将每个项目加入优先队列
for (int i = 0; i < m; i++) {
queue.offer(new Project(scan.nextLong(), 0, scan.nextLong()));
}
// 将n个人一一丢入项目中
for (int i = 0; i < n; i++) {
Project p = queue.poll(); // 弹出队头项目,即当前收益最大的项目
if (p.earn() <= 0) { // 若<=0说明没必要继续丢入项目了,收入已达到最大值
break;
}
cost += p.earn(); // 收益累加
queue.offer(new Project(p.k, p.x + 1, p.b)); // 将x+1后再加入队列
}
System.out.println(cost);
}
}
(by 归忆)