Codeforces Round #378 (Div. 2) D题（二分答案，贪心判断）

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int x[100100],y[100010],flag[100010];
int judge(int n,int m)//从前往后找是否可通过所有考试
{
    int sum=0;
    memset(flag,0,sizeof(flag));
    for(int i=n;i>=1;i--)
    {
        if(x[i]!=0&&!flag[x[i]])//没考过
        {
            flag[x[i]]=1;//标记
            sum+=y[x[i]];//加上考试要复习的时间
            m--;//考试门数减1
        }
        else if(sum>0)
        {
            sum--;//可复习就减少复习天数
        }
    }
    if(m!=0||sum!=0)
        return 0;
    else//所有的考完了，复习天数够
        return 1;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x[i]);
        }
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&y[j]);
        }
        int l=1,r=n;
        while(l<=r)//二分枚举答案
        {
            int mid=(l+r)>>1;
            if(judge(mid,m))
            {
                r=mid-1;
            }
            else
            {
                l=mid+1;
            }
        }
        if(!judge(n,m))//最大考不完输出-1
            printf("-1\n");
        else
            printf("%d\n",l);
    }
}

题意：n天要通过m科考试，d[i]每天可以考哪门科目，a[i]每门科目需要的准备时间，一天可以考试也可以准备，求最少时间。