题意:有n种动物要看病,每种动物有自己最大的看病次数(ai),但是医生只看k次,问看完k次后,剩下的要看病的动物。
分析:握草啊,我这个渣渣算是又领悟了一番二分查找的神奇!虽然之前有想过二分查找,但是苦于自己的思维太狭窄了。。根本没想过可以这么找啊。大神就是大神!要多多学习才是,具体讲解看代码(自己的理解而已。。)
Examples
input
3 3 1 2 1
output
2
input
4 10 3 3 2 1
output
-1
input
7 10 1 3 3 1 2 3 1
output
6 2 3
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define ll __int64
#define maxn 100005
#define maxa 1e9
#define maxk 1e14
ll n,k,a[maxn];
vector<ll> v;
int main()
{
while(~scanf("%I64d%I64d",&n,&k))
{
int i;
ll sum=0;
for(i=1;i<=n;i++)
{
scanf("%I64d",&a[i]);
sum+=a[i];
}
if(sum<k)
{printf("-1\n");continue;} //这步是自己加上去的,不知道有没有多余就是了
ll l,r,t=-1; //开始进行二分查找。在[1,1e9]之间找到满足小于k趟看病,循环到哪个最多次数的得病动物,用t记录该次数
l=1;r=maxa;
while(l<=r)
{
ll mid=(l+r)/2;
//int flag=0;
ll ans=0;
for(i=1;i<=n;i++)
{
ans=ans+min(a[i],mid); //记录到达该次数最多可以看病的趟数
}
if(ans>=k){
r=mid-1;t=mid; //关键步, 记录次数
}
else l=mid+1;
}
//cout<<t<<endl;
sum=0;ll p;
if(t==-1) printf("-1\n"); //如果没有找到就输出-1
else
{
for(i=1;i<=n;i++) sum+=min(a[i],t-1); //记录比最大次数少的趟数
for(i=1;i<=n;i++)
{
if(a[i]>=t)sum++;
if(sum==k){ //加上比最大次数大的,一直到趟数等于k,记录此时的位置p,表示位置p之前的都已经看过病,剩下的为位置p之后的
p=i;break;
}
}
// cout<<p<<endl;
v.clear();
for(i=p+1;i<=n;i++) if(a[i]>=t) v.push_back(i); //用vector记录剩下要看病的动物编号
for(i=1;i<=p;i++) if(a[i]>t) v.push_back(i);
for(i=0;i<v.size();i++)
i==0?printf("%I64d",v[i]):printf(" %I64d",v[i]);
printf("\n");
}
}
return 0;
}