问题 E: 买礼物的艰辛
题目描述
小 X 同学给小 C 同学选了 N 件礼物,决定顺序购买并赠送,但作为一个没有工资没有零花钱的可怜小朋友,有 M 位好心的同学伸出了援助之手,然而为了减少最高的借款量,小 X 同学希望 OI 竞赛的你为他合理规划,使得他能轻松快乐地送出礼物。
输入
第一行输入两个用空格隔开的正整数 N 和 M。
以下 N 行,每行一个不超过 10000 正整数,依次表示礼物的价格。
输出
一行一个整数,最高借款量。
样例输入 Copy
7 5
100
400
300
100
500
101
400
样例输出 Copy
500
提示
样例解释:
借第一个 500,够 1,2 个人,
然后借第二个 500,够 3,4 个人,下来不够第五个人,余下的钱全丢失。
借第三个 500,够第 5 个人。
借第四个 500,够第 6 个人,余下的钱全丢失。
借第五个 500,够第 7 个人,全都发到礼物,任务完成。
数据规模:
30%的数据满足:n≤10;
60%的数据满足:n≤1000;
100%的数据满足:n≤100000。
GPLv2 licensed by HUSTOJ 2020
开始以为用分堆做的,但是忽略了题干不能排序,
所以正确做法使二分。
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
priority_queue<int> dxq;//普通的优先级队列,按从大到小排序
priority_queue<int, vector<int>, greater<int> > xdq;//从小到大的优先级队列
int main()
{
int n,m;
cin>>n>>m;
for(int i=1; i<=n; i++)
{
ll a;
scanf("%lld",&a);
dxq.push(a);///
}
for(int i=1; i<=m; i++)
{
if(!dxq.empty())
{
ll a=dxq.top();
dxq.pop();
xdq.push(a);
}
}
while(!dxq.empty())
{
ll a=dxq.top();
dxq.pop();
ll b=xdq.top();
xdq.pop();
xdq.push(a+b);
}
ll ans=0;
while(!xdq.empty())
ans=xdq.top(),xdq.pop();
cout<<ans;
return 0;
}
/**************************************************************
Problem: 15682
User: 2019UPC110
Language: C++
Result: 答案错误
****************************************************************/
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
ll n,m;
ll a[100005];
priority_queue<int> dxq;//普通的优先级队列,按从大到小排序
priority_queue<int, vector<int>, greater<int> > xdq;//从小到大的优先级队列
bool judge(ll mid)
{
ll sum1=mid;
ll cnt=1;
for(ll i=0; i<n; i++)
{
if(sum1>=a[i])
sum1-=a[i];///够用
else
{
cnt++;
if(cnt>m)return 0;
sum1=mid-a[i];///不够时,再来个
}
}
return 1;
}
int main()
{
cin>>n>>m;
ll l=0,r=0;
for(ll i=0; i<n; i++)
{
cin>>a[i];
r=r+a[i];
l=max(l,a[i]);
}
while(l<=r)
{
ll mid=(l+r)/2;
if(judge(mid))
{
r=mid-1;
}
else
{
l=mid+1;
}
}
cout<<l<<endl;
return 0;
}
/**************************************************************
Problem: 15682
User: 2019UPC110
Language: C++
Result: 正确
Time:46 ms
Memory:2808 kb
****************************************************************/