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

题目2 : 优化延迟

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

小Ho编写了一个处理数据包的程序。程序的输入是一个包含N个数据包的序列。每个数据包根据其重要程度不同，具有不同的"延迟惩罚值"。序列中的第i个数据包的"延迟惩罚值"是Pi。如果N个数据包按照<Pi1, Pi2, ... PiN>的顺序被处理，那么总延迟惩罚

SP=1*Pi1+2*Pi2+3*Pi3+...+N*PiN（其中i1, i2, ... iN是1, 2, 3, ... N的一个排列)。

小Ho的程序会依次处理每一个数据包，这时N个数据包的总延迟惩罚值SP为

1*P1+2*P2+3*P3+...+i*Pi+...+N*PN。  

小Hi希望可以降低总延迟惩罚值。他的做法是在小Ho的程序中增加一个大小为K的缓冲区。N个数据包在被处理前会依次进入缓冲区。当缓冲区满的时候会将当前缓冲区内"延迟惩罚值"最大的数据包移出缓冲区并进行处理。直到没有新的数据包进入缓冲区时，缓冲区内剩余的数据包会按照"延迟惩罚值"从大到小的顺序被依次移出并进行处理。

例如，当数据包的"延迟惩罚值"依次是<5, 3, 1, 2, 4>，缓冲区大小K=2时，数据包被处理的顺序是：<5, 3, 2, 4, 1>。这时SP=1*5+2*3+3*2+4*4+5*1=38。

现在给定输入的数据包序列，以及一个总延迟惩罚阈值Q。小Hi想知道如果要SP<=Q，缓冲区的大小最小是多少？

输入

Line 1: N Q

Line 2: P1 P2 ... PN

对于50%的数据： 1 <= N <= 1000

对于100%的数据： 1 <= N <= 100000, 0 <= Pi <= 1000, 1 <= Q <= 1013

输出

输出最小的正整数K值能满足SP<=Q。如果没有符合条件的K，输出-1。

样例输入

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;
}