HDU 4546 比赛难度（优先队列）

最新推荐文章于 2017-07-19 09:57:45 发布

原创最新推荐文章于 2017-07-19 09:57:45 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

数据结构-时空优化专栏收录该内容

23 篇文章

订阅专栏

本文介绍了解决HDU4546竞赛题目的方法。通过将数列排序并使用优先队列来维护当前和加下一个位置和的最小值，确保每次加入队列的值都是最小的。最终实现O(m log m)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

HDU 4546 比赛难度

题目链接

思路：由于m不是很大，如果用一个优先队列维护，如果每次能保证加入的值是最小的，那么只需要加入m次就能完成了，时间复杂度足够，那么如何保证呢，就把数列排序，维护优先队列为当前和加下一个位置和的最小值，每次一个出队，把下一个位置取于不取在入队，最后求出答案即可

代码：

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;

const int N = 10005;

int t, n, m, a[N];

struct Node {
	int sum, nextsum, nextid;
	Node() {}
	Node(int sum, int nextsum, int nextid) {
		this->sum = sum;
		this->nextsum = nextsum;
		this->nextid = nextid;
	}
	bool operator < (const Node &c) const {
		return nextsum > c.nextsum;
	}
};

const int INF = 0x3f3f3f3f;

int main() {
	int cas = 0;
	scanf("%d", &t);
	while (t--) {
		scanf("%d%d", &n, &m);
		for (int i = 0; i < n; i++) scanf("%d", &a[i]);
		sort(a, a + n);
		priority_queue<Node> Q;
		Q.push(Node(0, a[0], 0));
		int cnt = 0;
		while (!Q.empty()) {
			Node u = Q.top();
			Q.pop();
			if (u.nextid >= n) continue;
			Q.push(Node(u.sum, u.sum + a[u.nextid + 1], u.nextid + 1));
			Q.push(Node(u.nextsum, u.nextsum + a[u.nextid + 1], u.nextid + 1));
			cnt++;
			if (cnt == m) {
				printf("Case #%d: %d\n", ++cas, u.nextsum);
				break;
			}
		}
	}
	return 0;
}