CodeForces - 732D Exams(二分+贪心)
题意:要在n天内通过m科考试,考每一科目是,都必须要有一定的准备时间,准备时间可以不连续,给出n天内每一天可以考的
科目,输出最少需要多少天才能通过所有的科目。
思路:对需要多少天才能通过所有科目进行二分,关键是如何判断在这些天是否能通过所有科目的考试,贪心来求,从后向前遍
历,因为越往后这一科目准备的时间就越长,当遇到没有考的科目所需准备所有考试的天数need就加上这一科目需要准备的时间,
当在这一天考试的科目已考完或者不能进行考试时,就可以准备考试了,所以need-1,最后判断所有科目是否都考过和need是否为
0就行了
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <math.h>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
using namespace std;
const int MAXN=100005;
int a[MAXN],d[MAXN];
bool vis[MAXN];
int n,m;
bool pass(int num)
{
for(int i=1;i<=m;i++)
vis[i]=false;
int need=0;
for(int i=num;i>=1;i--)
if(!vis[a[i]]&&a[i])
{
need+=d[a[i]];
vis[a[i]]=true;
}
else if(need>0) need--;
for(int i=1;i<=m;i++)
if(!vis[i])
return false;
if(!need)
return true;
return false;
}
int main(void)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=m;i++)
scanf("%d",&d[i]);
int l=1,r=n;
int mid;
int ans=-1;
while(l<=r)
{
mid=(l+r)/2;
if(pass(mid))
{
r=mid-1;
ans=mid;
}
else l=mid+1;
}
printf("%d\n",ans);
return 0;
}