Codeforces-Problem 732D Exam

题目：

http://codeforces.com/problemset/problem/732/D

题目大意：

有n天，m门科目（接下来分别输入n天内每天可以干的事情，如果输入1则表示可以考科目1，2则表示可以考科目2，以此类推。0表示只可以复习不可以考试。考试的那天不可以复习）（接下来输入m个数，第i个数表示第i门科目需要复习的时间）。问把所有科目考完最快是第几天。

#include<iostream>
#include<string.h>
using namespace std;
int num[100100],pre[100100];
bool flag[100100];
int n,m;
bool judge(int day){
memset(flag,0,sizeof(flag));
int rest=0; int cnt=0;
for(int i=day;i>=1;i--){    //从后往前找
if(num[i]!=0&&flag[num[i]]==0){
flag[num[i]]=1;     //记录这门科目是否考过
cnt++;              //记录考过的总门数
rest+=pre[num[i]];  //需要复习的总时间
}
else if(rest!=0)        //如果rest为0就不减了，即可以有某些日子颓着啥也不干，否则rest会出现负的
rest--;

}
if(cnt==m&&rest==0)         //如果考过的门数等于科目总数，并且需要复习的时间为0（已经在从后往前跑循环的时候被减没了）
return true;
return<
4000
/span> false;

}
int main(){
int i,left,right,mid;
while(cin>>n>>m){
for(i=1;i<=n;i++)
cin>>num[i];
for(i=1;i<=m;i++)
cin>>pre[i];
left=1;right=n;
while(left<right){     //二分查找
mid=left+(right-left)/2;
if(judge(mid)){   //如果中间的那一天符合条件，就向左找有没有更小的
right=mid;
}
else left=mid+1;  //否则往右找

}
if(judge(right))      //存在符合条件的
cout<<right<<endl;  //输出位置
else cout<<"-1"<<endl;

}

return 0;
}