天天看点

hihocoder1269优化延迟

这道题其实只是一道很裸的二分+优先队列,然而本人比较笨,所以WA了好多次,下面附上代码

#include <iostream>
#include <queue>
#include <string.h>
using namespace std;
priority_queue<long long>que;
int n,t=;
long long q;
int a[];
long long sum(int k)
{
    int b[];
    memset(b,,sizeof(b));
    long long sum2=;
    int i=,j=;
    while(i<=n)
    {
        if(que.size()<k)
        {
            que.push(a[i]);
            i++;
        }
        else
        {
            b[j++]=que.top();
            que.pop();
            que.push(a[i++]);
            //每次弹出一个数值后,一定要再压入一个,否则的话,一个数会用到两个i,会溢出
        }
    }
    while(!que.empty())
    {
        b[j++]=que.top();
        que.pop();
    }
    for(int i=; i<=n; i++)
        sum2+=i*b[i];
        //这里用数组会比较方便一点,当然你每弹出一个乘一次也是可以的,但是可能会比较麻烦一点
    return sum2;
}

int main()
{
    while(cin>>n>>q)
    {
        for(int i=; i<=n; i++)
            cin>>a[i];
        int left=;
        int right=n;
//二分这块的处理很重要,很多人觉得简单,但是很容易写错

        while(left<=right)//一定要有等号
        {
            int mid=(left+right)/;
            if(sum(mid)<=q)//一定要有等号
            {
                t=min(mid,t);
                //之所以用二分是因为这道题随着k的增加,和逐渐减少,是递减的,但又不是严格意义上的递减有可能会出现相等的地方,所以这里要用一下min
                right=mid-;
            }
            else
            {
                left=mid+;
            }
        }
        if(sum(t)<=q)
            cout<<t<<endl;
        else cout<<-<<endl;
//下面的这种也是可以的,但是我觉得并不是很好理解
        /*while(left<=right)
                {
                    int mid=(left+right)/2;
                    if(sum(mid)<=q)
                    {
                        right=mid-1;
                    }
                    else if(sum(mid)>q)
                    {
                        left=mid+1;
                    }
                }
        cout<<flag<<endl;
        if(sum(left)<=q)
        cout<<left<<endl;
        else cout<<-1<<endl;*/

    }
    return ;
}