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