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