codeforces G. Doctor（二分查找）

最新推荐文章于 2024-01-12 21:37:39 发布

原创最新推荐文章于 2024-01-12 21:37:39 发布 · 398 阅读

0 ·

CC 4.0 BY-SA版权

二分查找专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个使用二分查找算法解决动物看病次数问题的程序设计案例。通过二分查找确定医生在限定次数内能治疗哪些种类的动物，最终输出未被治疗的动物编号。

题意：有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;
}