传送门
思路:
- 现有m个人和n天的课表,要求安排一个可行方案,满足第j天有d[j]个人工作,第i个人总共工作w[i]天。且每个人一开始工作就必须连续工作w天,然后至少休息h天才能工作。如果有可行解第一行输出"1",然后输出m行,每i行含w[i]/w个数,表示第i个人每次工作的开头那天;无解直接输出"-1"。
- 从头按顺序开始安排每一天工作的任选,首选工作量最大的且可以工作(既没有在进行上一次工作,也休息了至少h天;且后w天都需要人工作)的人。
- 这个时候就用一个优先队列来存m个人的当前工作量情况,且用一个队列过渡。
- 详细见代码。
代码实现:
#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll mod = 1e9 + 7;
const int N = 2005;
int m, n, w, h, ww, d[N];
vector<int> a[N];
struct cmp{ //优先队列重载
bool operator()(const pii p1, const pii p2){
return p1.first < p2.first;
}
};
priority_queue <pii, vector<pii>, cmp> qw;
bool check(int bg, int id){
//如果这个人当前还不能工作
if(a[id].size() && a[id].back()+w+h-1>=bg) return 0;
//如果当前天已经不再需要人工作
for(int i = bg; i < bg+w; i ++)
if(d[i] <= 0) return 0;
return 1;
}
signed main()
{
IOS;
cin >> m >> n >> w >> h;
bool ok = 1;
for(int i = 1; i <= m; i ++){
cin >> ww; qw.push({ww, i});
if(ww%w) ok = 0;
}
for(int i = 1; i <= n; i ++) cin >> d[i];
// while(qw.size()){cout << qw.top().first << endl; qw.pop();}
if(!ok) cout << -1 << endl; //如果某个人的初始工作量不能整除w,就不可能实现正好w[i]的工作量
else{
ok = 1;
for(int i = 1; i <= n; i ++){
if(!d[i]) continue;
queue<pii> q;
while(qw.size()&&d[i]){
int id = qw.top().second, da = qw.top().first; qw.pop();
if(check(i, id)){ //如果此人当前可以工作,且后w天也需要人工作
// cout << "i:" << i << " " << "di:" << d[i] << endl;
a[id].push_back(i);
q.push({da-w, id}); //更新此人的工作量
for(int j = i; j < i+w; j ++) d[j] --; //后w天需要工作的人数都-1
}
else q.push({da, id}); //如果不能工作就直接进行下一轮筛选
}
// cout << "i:" << i << " " << "di:" << d[i] << endl;
if(d[i]){ok = 0; break;} //如果所有人都安排完了却依旧不能满足当天所需的工作人数
while(q.size()){ //将临时队列q里的数据转入优先队列qw中
qw.push(q.front());
q.pop();
}
}
while(qw.size()){
if(qw.top().first){
ok = 0; //如果最好还有某一天所需工作人数不够
// cout << "id:" << qw.top().second << " " << "id:" << qw.top().first << endl;
}
qw.pop();
}
if(!ok) cout << -1 << endl;
else{
cout << 1 << endl;
for(int i = 1; i<= m; i ++)
for(int j = 0; j < a[i].size(); j ++)
cout << a[i][j] << " \n"[j == a[i].size()-1];
}
}
return 0;
}