題目大意:
有m個bug,n個學生,每個bug有自己的複雜值,每個學生有自己的能力值,只能解決複雜值小於自己能力值的bug,且每天只能解決一個bug,但是每個學生處理bug需要報酬,問在報酬不超過s的情況下,在最短的時間內,能將bug全部解決完。
做法:
首先明確一點,如果能夠解決所有bug,那麼解決的天數一定在1~m之間。先判斷m天能不能解決,如果不能則輸出NO,如果可以再進行下面的操作。
我們可以二分枚舉解決問題所需的天數t,將bug的複雜值和同學的能力值按照從大到小的順序排序。首先先處理最複雜的bug,將能夠解決該bug的同學添加到優先隊列,讓這些同學中要錢最少的一個來解決這些bug,顯然,這個同學最多能解決t個bug。然後將該同學彈出隊列。再用相同的操作處理剩下的沒有解決的bug。如果在過程中隊列爲空,或者花費大於s,那麼直接返回。
代碼:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#define N 100010
using namespace std;
struct aa
{
int sk,f,id;
bool operator <(const aa &s)const
{
return sk<s.sk;
}
}stu[N],bug[N];
struct cmp{
bool operator()(aa &x,aa &y)
{
return x.f>y.f;
}
};
int n,m,s,ans[N];
bool check(int t)
{
int cur=0;
priority_queue<aa,vector<aa>,cmp> que;
for(int i=m-1,j=n-1;i>=0;i-=t)
{
for(;j>=0 && stu[j].sk>=bug[i].sk;j--) que.push(stu[j]);
if(que.empty()) return false;
cur+=que.top().f;
if(cur>s) return false;
for(int k=i;k>=max(0,i-t+1);k--)
ans[bug[k].id]=que.top().id;
que.pop();
}
return true;
}
int main()
{
scanf("%d%d%d",&n,&m,&s);
for(int i=0;i<m;i++) scanf("%d",&bug[i].sk),bug[i].id=i;
for(int i=0;i<n;i++) scanf("%d",&stu[i].sk);
for(int i=0;i<n;i++) scanf("%d",&stu[i].f),stu[i].id=i;
sort(stu,stu+n);
sort(bug,bug+m);
if(!check(m)) {printf("NO");return 0;}
int lb=0,ub=m+1;
while(ub-lb>1)
{
int mid=(ub+lb)/2;
if(check(mid))
ub=mid;
else lb=mid;
}
printf("YES\n");
check(ub);
for(int i=0;i<m;i++)
cout<<ans[i]+1<<" ";
return 0;
}