Codeforces Round #378 (Div. 2) D題(二分答案,貪心判斷)
阿新 • • 發佈:2018-12-23
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int x[100100],y[100010],flag[100010];
int judge(int n,int m)//從前往後找是否可通過所有考試
{
int sum=0;
memset(flag,0,sizeof(flag));
for(int i=n;i>=1;i--)
{
if(x[i]!=0&&!flag[x[i]])//沒考過
{
flag[x[i]]=1;//標記
sum+=y[x[i]];//加上考試要複習的時間
m--;//考試門數減1
}
else if(sum>0)
{
sum--;//可複習就減少複習天數
}
}
if(m!=0||sum!=0)
return 0;
else//所有的考完了,複習天數夠
return 1;
}
int main()
{
int n,m;
while (~scanf("%d%d",&n,&m))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&x[i]);
}
for(int j=1;j<=m;j++)
{
scanf("%d",&y[j]);
}
int l=1,r=n;
while(l<=r)//二分列舉答案
{
int mid=(l+r)>>1 ;
if(judge(mid,m))
{
r=mid-1;
}
else
{
l=mid+1;
}
}
if(!judge(n,m))//最大考不完輸出-1
printf("-1\n");
else
printf("%d\n",l);
}
}
題意:n天要通過m科考試,d[i]每天可以考哪門科目,a[i]每門科目需要的準備時間,一天可以考試也可以準備,求最少時間。