首页 > 代码库 > HDU4546 比赛难度 (优先队列)
HDU4546 比赛难度 (优先队列)
题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=4546
题意:
给你n个数,问这n个数的组合的和,第m小是多少
分析:
对a[]序列进行排序
我们用一个结构体来存储 当前的和与它马上要加上的一个值将其放入队列中,
那么我们用队首的元素加上队首元素要加的值得下一个放入队列中,弹出队首元素,
进行m次这样的操作就可以得到第m小值.
代码如下:
#include <iostream> #include <cstring> #include <cstdio> #include <queue> #include <algorithm> using namespace std; const int maxn = 10010; int a[maxn]; struct elg{ int start,i; elg(int a,int b):start(a),i(b){} bool operator < (const struct elg &tmp)const{ return start + a[i] > tmp.start + a[tmp.i]; } }; int main() { int n,m,t,cas=1; 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<elg> Q; Q.push(elg(0,0)); int ans = 0; while(m--){ if(Q.empty()) continue; elg tmp = Q.top(); ans = tmp.start +a[tmp.i++]; Q.pop(); if(tmp.i<n){ Q.push(tmp); tmp.start = ans; Q.push(tmp); } } printf("Case #%d: %d\n",cas++,ans); } return 0; }
HDU4546 比赛难度 (优先队列)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。