HDU4546(组合求第k小优先级队列)

最新推荐文章于 2022-10-09 10:04:34 发布

UFO___

最新推荐文章于 2022-10-09 10:04:34 发布

阅读量547

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签：算法

本文链接：https://blog.youkuaiyun.com/UFO___/article/details/76051191

算法专栏收录该内容

48 篇文章

订阅专栏

博客探讨了如何使用优先级队列解决HDU4546问题，提供了巧妙的解题模板，虽然作者表示对这种方法的理解有待深入。

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

这个直接当模板用吧，太巧妙了，我不是很能理解。

#include<iostream>
        #include<algorithm>
        #include<queue>
        #include<cstdio>
        using namespace std;
        
        int arr[10005];
        
        typedef struct node
        {
        	int sum;
        	int next_id;
        	friend bool operator < (node a,node b)
        	{
        		return a.sum+arr[a.next_id]>b.sum+arr[b.next_id];
			}
		}node;
        
        int main()
{
	    int T;
	    scanf("%d",&T);
	    for(int Case=1;Case<=T;Case++)
	    {
	    	int n,m;
	    	scanf("%d%d",&n,&m);
	    	for(int i=0;i<n;i++) scanf("%d",&arr[i]);
	    	sort(arr,arr+n);
	    	priority_queue<node> pq;
	    	pq.push(node{0,0});
	    	int ans=0;
	    	while(m--)
	    	{
	    		if(pq.empty()) break;
	    		if(pq.top().next_id+1<n)
	    		{   
	    			node first=pq.top();//马上要入队列的第一个元素 
	    		    first.next_id++;
	    		    node second=pq.top();//马上要入队列的第二个元素
	    		    second.sum+=arr[second.next_id];
	    		    second.next_id++; 
					pq.push(first);
				    pq.push(second);   
				}
				ans=pq.top().sum+arr[pq.top().next_id];
	    		pq.pop();
			}
			printf("Case #%d: %d\n",Case,ans);
		}
		return 0;
}