https://codeforces.com/gym/101987/attachments
题意:m个人, n天, 每个人一旦开始工作需要连续工作w天, 工作完w天后需要休息h天, 每天需要
d
i
d_i
di个人, 每个人要求工作
W
i
W_i
Wi天, 求方案。
思路:每一天都要安排好,那么我们一天一天来安排,对于第i天,每个人来劳动都是一样的效果,但是【一个人不能同时多次劳动以及连续劳动之间需要休息】【不同的人可以同时劳动】,因此优先让剩余待劳动天数多的人来劳动,用优先队列来维护。
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
int m,n,W,h,w[3000],d[3000];
int c[3000],sum,s1,s2;
struct Node{
int id,cnt;
bool operator < (const Node x)const{
return cnt<x.cnt;
}
};
priority_queue<Node> Q;
vector<P> add[3000];
vector<int> ans[3000];
int main()
{
cin>>m>>n>>W>>h;
for(int i=1;i<=m;i++)scanf("%d",&w[i]),Q.push((Node){i,w[i]}),s1+=w[i];
for(int i=1;i<=n;i++)scanf("%d",&d[i]),s2+=d[i];
if(s1!=s2)
{
puts("-1");
return 0;
}
for(int i=1;i<=n;i++)
{
for(int j=0;j<add[i].size();j++)Q.push((Node){add[i][j].first,add[i][j].second});
sum+=c[i];
if(sum>d[i])
{
puts("-1");
return 0;
}
while(sum<d[i])
{
sum++;
c[i+W]--;
if(Q.empty())
{
puts("-1");
return 0;
}
Node x=Q.top();
Q.pop();
if(x.cnt>=2*W)add[i+W+h].push_back(make_pair(x.id,x.cnt-W));
ans[x.id].push_back(i);
}
}
puts("1");
for(int i=1;i<=m;i++)
{
for(int j=0;j<ans[i].size();j++)printf("%d ",ans[i][j]);
puts("");
}
return 0;
}