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