补题J:Flowers
题目链接
tot(为花的贡献和)
本题给出数据n,m,n为花卉种类,m为每束花应该有m种,check中的x为x束花,假设有x束花,则应该有xm朵花,在n种花中,(check过程)如果a[i]>x则说明花的数量大于花束的数量,这种花可以用在每一种,而且还多出来,tot中加上x,反之第i种花的数量小于花束的数量则tot应该加上该种花的数量(a[i])。如果tot最终大于xm则花的贡献量大于应该有的花数,则check函数返回1(最佳答案大于等于x也就是传过去的mid),使ans值等于mid并使左边界等于mid+1,反之如果tot<x*m,则说明花的贡献量小于应有的花数,则check返回0(最佳答案小于等于x),使右边界等于mid-1。(二分过程)。
代码:
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<set>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<algorithm>
#include<sstream>
#include<memory>
#include<functional>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a));
#define ll long long int
const int INF = 0x3f3f3f3f;
int n,m;
ll a[300030];
bool check(ll x)//x束
{
ll tot=0;//tot是花的贡献和
for(int i=0;i<n;++i)
tot+=min(a[i],x);//a[i]>=x,该花可以用在每一束,则取x,a[i]<x,该花不能用在每一束,取该花的数量(该种花需要全部用完)
if(tot>=x*m)
return 1;
return 0;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
ll sum=0;
for(int i=0;i<n;++i)
{
scanf("%d",&a[i]);
sum+=a[i];
}
ll l=0,r=sum/m,ans;//r是右边界
while(l<=r)
{
ll mid=(l+r)>>1;
if(check(mid))
{
l=mid+1;
ans=mid;
}
else
r=mid-1;
}
cout<<ans<<endl;
}
return 0;
}