CodeForces 83B Doctor

二分查找。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;

const int maxn=100000+10;
int n;
long long a[maxn],k;
vector<int>ans;

bool check(long long m)
{
    long long res=0;
    for(int i=1;i<=n;i++) res=res+min(m,a[i]);
    if(res>=k) return 1;
    return 0;
}

int main()
{
    scanf("%d%lld",&n,&k);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    if(k==0)
    {
        for(int i=1;i<=n;i++) printf("%lld ",a[i]);
        printf("
");
        return 0;
    }
    long long l=1,r=1e9, p=-1;
    while(l<=r)
    {
        long long mid=(l+r)/2;
        if(check(mid)) p=mid,r=mid-1;
        else l=mid+1;
    }
    if(p==-1) printf("-1
");
    else
    {
        long long sum=0;
        int pos;
        for(int i=1;i<=n;i++) sum=sum+min(a[i],p-1);
        for(int i=1;i<=n;i++)
        {
            if(a[i]>=p) sum++;
            if(sum==k) { pos=i; break; }
        }

        for(int i=pos+1;i<=n;i++) if(a[i]>=p) ans.push_back(i);
        for(int i=1;i<=pos;i++) if(a[i]>=p+1) ans.push_back(i);

        for(int i=0;i<ans.size();i++) printf("%d ",ans[i]);
        printf("
");

    }
    return 0;
}