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