这道题其实只是一道很裸的二分+优先队列,然而本人比较笨,所以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 ;
}