Codeforces 83B Doctor(二分)

Codeforces 83B Doctor

题意：

有n个人去看病，医生只看k次就回家，而每个人可能需要看多次病，看完一次就要到队尾去排队，问看k次病后队列是什么样的。

分析：

二分找每个人最多看多少次后结束，去掉看完病的，重新排队列输出。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define mod 1e9+7
#define for(i,nn,mm) for(int i=nn;i<=mm;i++)
typedef long long LL;
const int maxn=100005;
LL n,k,num[maxn],l,r,sum,mid,x[maxn],y[maxn],ans,xs,ys;
int main(){
	scanf("%I64d%I64d",&n,&k);
	sum=0;
	r=0;
	for(i,0,n-1){
		scanf("%I64d",&num[i]);
		sum+=num[i];
		if(num[i]>r)r=num[i];
	}
	if(sum<k)printf("-1\n");
	else if(sum>k){
		l=0;
		while(l<r){
			mid=(l+r)/2;
			ans=mid*n;
			for(i,0,n-1)if(num[i]<mid)ans-=mid-num[i];
			if(ans>k)r=mid;
			else l=mid+1;
		}
		l--;
		ans=l*n;
		for(i,0,n-1)if(num[i]<l)ans-=l-num[i];
		ans=k-ans;
		xs=0;
		ys=0;
		for(i,0,n-1){
			if(num[i]==l+1){
				if(ans)ans--;
				else y[ys++]=i;
			}
			else if(num[i]>l+1){
				if(ans){
					x[xs++]=i;
					ans--;
				}
				else y[ys++]=i;
			}
		}
		printf("%I64d",y[0]+1);
		for(i,1,ys-1)printf(" %I64d",y[i]+1);
		for(i,0,xs-1)printf(" %I64d",x[i]+1);
		printf("\n");
	}
	return 0;
}