L. Working Plan (优先队列贪心思维课表安排) ICPC Seoul 2018

本文主要是介绍L. Working Plan (优先队列贪心思维课表安排) ICPC Seoul 2018，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

在这里插入图片描述 传送门
思路：

现有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;
}