题意:
有n个人要去看医生,医生只看k次就回家,先给出每个人要看医生的次数,看完一次后如果需要再看,则排到队尾去,问你最后队列的情况是怎么样的。
思路:
一开始以为是像利用Havel-Hakimi定理去求是否为无向图那样不断模拟,但是如果数据是1~1e5的话就会变成n^2logn的复杂度,太高了。
所以我们可以通过二分法来获得最后每个人最多看l次,即最后看完的那个人的所看的次数,然后通过这个次数来模拟一遍,得到最后看完医生的那个人是谁,那么排在那个人之后还需要看医生的人就排到队头,它之前还需要看医生的人就接在队尾。
#include<cstdio>
#include<algorithm>
using namespace std;
typedef __int64 LL;
const int MAX=1e5+5;
int n,ans[MAX];
LL k;
struct Node{
LL num;
}a[MAX];
bool ok(LL mid){
LL res=0;
for(int i=1;i<=n;i++){
res+=min(mid,a[i].num);
}
if(res>=k) return true;
else return false;
}
int main(){
scanf("%d%I64d",&n,&k);
LL sum=0;
for(int i=1;i<=n;i++){
scanf("%I64d",&a[i].num);
sum+=a[i].num;
}
if(k==0){
for(int i=1;i<=n;i++){
if(i!=1) printf(" ");
printf("%d",i);
}
printf("\n");
}
else{
if(sum<k) printf("-1\n");
else if(sum==k){
printf("\n");
}
else{
LL l=1,r=1e9;
while(l<r){
LL mid=l+(r-l)/2;
if(ok(mid)) r=mid;
else l=mid+1;
}
//printf("%I64d\n",l);
LL res=0;
for(int i=1;i<=n;i++){
res+=min(a[i].num,l-1);
}
int st;
for(int i=1;i<=n;i++){
if(a[i].num>=l) res++;
if(res==k){
st=i;
break;
}
}
int cnt=0;
for(int i=st+1;i<=n;i++){
if(a[i].num>=l){
ans[cnt++]=i;
}
}
for(int i=1;i<=st;i++){
if(a[i].num>l){
ans[cnt++]=i;
}
}
for(int i=0;i<cnt;i++){
if(i!=0) printf(" ");
printf("%d",ans[i]);
}
printf("\n");
}
}
return 0;
}