CodeForces 732 D.Exams（二分+贪心）

Description
n天m场考试，每天可以准备一些考试或闲着，也可以去通过第d[i]场考试(d[i]!=0)，第i场考试需要提前准备a[i]天，问通过所有考试所需的最少天数，如果n天都考不完则输出-1
Input
第一行两整数n和m分别表示天数和考试数，之后n个数d[i]表示第i天可以考试的编号(d[i]=0则表示该天不能考试)，最后m个整数a[i]表示第i场考试需要提前准备的天数(1<=n,m<=100000,0<=d[i]<=m,1<=a[i]<=100000)
Output
输出通过所有考试的最少天数，如果n天都通不过所有考试则输出-1
Sample Input
7 2
0 1 0 2 1 0 2
2 1
Sample Output
5
Solution
二分答案，对每次的二分值mid，从后往前贪心的考试即可
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;
#define INF 0x3f3f3f3f
#define maxn 111111
int n,m,d[maxn],a[maxn],vis[maxn];
bool check(int x)
{
    memset(vis,0,sizeof(vis));
    int num=m;
    for(int i=x;i>=1;i--)
    {
        if(d[i]==0||vis[d[i]])continue;
        if((i-1)-(num-1)>=a[d[i]])
            num--,vis[d[i]]=1;
        else return 0;
        if(num==0)return 1;
    }
    if(num)return 0;
} 
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        int l=m,r=n,ans=-1,mid;
        for(int i=1;i<=n;i++)scanf("%d",&d[i]);
        for(int i=1;i<=m;i++)scanf("%d",&a[i]),l+=a[i];
        while(l<=r)
        {
            mid=(l+r)>>1;
            if(check(mid))r=mid-1,ans=mid;
            else l=mid+1;
        }
        printf("%d\n",ans);
    }
    return 0;
}