Codeforces Round #377 (Div. 2)D(二分)

题目链接：http://codeforces.com/contest/732/problem/D

题意：

在m天中要考k个课程，

数组a中有m个元素，表示第a[i]表示第i天可以进行哪门考试，若a[i]为0，则表示当天不能参加任何科目的考试，只能预习或者休息；

数组b中有k个元素，b[i]表示科目i需要复习几天才能通过；

问最快需要几天能通过所有考试，如果不能完成所有科目考试，输出-1；

 

思路：

用二分方法直接找满足条件的最小的数，其中数是指当前天是第几天，条件是指当前天以前是否能考完所有科目（包括当前天）;

 

代码：

#include <bits/stdc++.h>
#define MAXN 100000+10
using namespace std;

int a[MAXN], b[MAXN], n, k;

int gg(int x){  //××××××判断花费x天能否完成所有科目考试，即能否在第x天以前完成所有科目考试；
    int cnt=0, vis[MAXN];
    memset(vis, 0, sizeof(vis));
    for(int i=x; i>=0; i--){
        // cout << cnt << "//" << endl;
        if(!a[i]){
            cnt--;
            cnt=max(cnt, 0);
        }else if(!vis[a[i]]){
            cnt+=b[a[i]-1];
            // cout << cnt << "***" << endl;
            vis[a[i]]=1;
            if(cnt>i){
                return 0;
            }
        }else if(vis[a[i]]){
            cnt--;
            cnt=max(0, cnt);
        }
    }
    for(int i=1; i<=k; i++){
        if(!vis[i]){
            return 0;
        }
    }
    return 1;
}

int main(void){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n >> k;
    for(int i=0; i<n; i++){
        cin >> a[i];
    }
    for(int i=0; i<k; i++){
        cin >> b[i];
    }
    int l=0, r=n;
    while(l<r){   //××××××二分查找满足条件的最小数
        int mid=(l+r)>>1;
        // cout << mid << " " << gg(mid) << endl;//****
        if(gg(mid)){
            r=mid;
        }else{
            l=mid+1;
        }
    }
    if(l>=n){
        cout << "-1" << endl;
    }else{
        cout << l+1 << endl;
    }
    return 0;
}