[Offer收割]编程联赛1-B 优化延迟

描述

输入

输出

5 38
5 3 1 2 4

2

思路：
本问题实质上priority_queue（优先队列）问题，通过priority_queue的基本操作，将队列中最大值取出即可，STL中priority_queue是由max_heap（大根堆）实现，因为大根堆的数据结构刚好符合要求。本问题的重点在于能否想到使用priority_queue数据结构。
代码解释：

源代码中inp数组保存用户原始输入数据，test数组用于保存出队列后的数据，kq为priority_queue数据结构对象。
函数checkqueue的逻辑为：第一步生成test数据，先判断队列中的元素个数是否达到出队要求，若满足，则将顶部元素放入test中，并pop操作
若不满足则什么都不做，最后将inp中数据入队。第二部当inp中元素全部入队后，for循环结束，此时将kq队列中剩余元素一次复制到test中，并执行pop操作。最后计算结果。
源码：

#include <iostream>#include <queue>using namespace std;int N,Q;//N是数据量大小，Q是阈值。int inp[1000],test[1000];priority_queue<int> kq;bool checkqueue(int size){int cur=0;for(int i=0;i<N;i++)//将inp数组按题目要求处理{if(kq.size()==static_cast<unsigned>(size)&&!kq.empty())//当队列满时，取出最大权值数{test[cur++]=kq.top();kq.pop();}kq.push(inp[i]);//向队列中添加元素}while(!kq.empty())//将队列剩余元素清空{test[cur++]=kq.top();kq.pop();}int res=0;for(int i=0;i<N;i++){res += (i+1)*test[i];//统计结果}return res<=Q;}int main(){cin >> N >> Q;for(int i=0;i<N;i++){cin >> inp[i];}for(int i=2;i<=N;i++){if(checkqueue(i)){cout << i << endl;break;}}return 0;}