GYM 101755 K.Video Reviews（二分+贪心）

Description

一家公司想让$n$个人给他们的产品评论，所以依次去找这$n$个人，第$i$个人会评论当且仅当已经有$a_i$个人评论或他确实对这个产品感兴趣，但是这$n$个人都不对这个产品感兴趣，问这个公司至少要说服几个人对该产品该兴趣才能至少收到$m$个人的评论

Input

第一行两个整数$n,m$，之后输入$n$个整数$a_1,...,a_n$$(1\le m\le n\le 2\cdot 10^5,0\le a_i\le 10^5)$

Output

输出至少要说服的顾客数量

Sample Input

7 4
2 1 3 3 4 2 3

Sample Output

1

Solution

二分答案，假设二分值为$k​$，那么从第一个人开始，对于第$i​$个人，只要没有$a_i​$个人评论且还有说服名额就说服该人，因为说服一个人对评论数贡献为$1​$，而提早做贡献可以带来更多的贡献，以此求出得到的评论数与$m​$作比较来改变二分方向即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=200005;
int n,m,a[maxn];
bool check(int k)
{
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        if(a[i]<=ans)ans++;
        else if(k)ans++,k--;
        if(ans==m)return 1;
    }
    return 0;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    int l=0,r=m,mid,ans=m;
    while(l<=r)
    {
        mid=(l+r)/2;
        if(check(mid))r=mid-1,ans=mid;
        else l=mid+1;
    }
    printf("%d\n",ans);
    return 0;
}