cf Exams 贪心+二分

n天,有m门考试

---

10 3
0 0 1 2 3 0 2 0 1 2
1 1 4

---

为0则代表当天只可复习,1~m则代表可以参加某一门考试,不参加也可以复习.
每一门考试有自己需要的复习时间.
这题的做法是二分后,贪心检测.
实在是不知道怎么做..
因为这个题目是某种”分开的”,也就是你直接二分,二分后直接贪心,而不是他们混杂在一起,这一点我确实没有想到..这里的贪心,不像一些很难的贪心思路,可以直接得到最优解,而只是很大概的检测一下最差情况,这也是我没有想到的.以后要注意

#include <iostream>
#include <set>
using namespace std;
#define debug(x) std::cerr << #x << " = " << (x) << std::endl
typedef long long LL;
const int MAXN = 1e5+17;
int d[MAXN],a[MAXN],m;
bool check(int p)
{
    int need = 0,gt = 0;
    set<int > s;
    for (int i = p; i > 0; --i)
    {
        if(d[i]!=0&&s.find(d[i])==s.end())
            s.insert(d[i]),need+=a[d[i]],gt++;
        else if(need>0)
            need--;
    }   
    return need<=0&&gt==m;
}
int main(int argc ,char const *argv[])
{
     #ifdef GoodbyeMonkeyKing
    freopen("Input.txt","r",stdin);freopen("Output.txt","w",stdout);
    #endif
    int n;
    cin>>n>>m;
    for (int i = 1; i <= n; ++i)
        cin>>d[i];
    for (int i = 1; i <= m; ++i)
        cin>>a[i];
    int l = -1,r = n+1;
    while(r-l>1)
    {
        int mid = (l+r)/2;
        if(check(mid))
            r = mid;
        else
            l = mid;
    }
    cout<<((r>n)?-1:r)<<endl;
    return 0;
}