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