【蓝桥杯3541】最大开支（优先队列&java）

问题描述

输入输出

 

解题思路

根据题意知，当前项目的花费为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 归忆）