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);
}
}