POJ 1456 贪心 + 优先队列 / 并查集

题意

传送门 POJ 1456 Supermarket

题解

贪心 + 优先队列

使用贪心策略，对于每个时间点 $t$，在保证不卖出过期商品的情况下，尽量卖出利润前 $t$ 大的商品。按照过期时间从小到大对商品排序，顺序考虑商品，使用优先队列动态维护满足上述性质的方案。

对于每一个扫描到的商品，若其过期时间大于当前堆中的商品数（此时之前扫描过的商品因为最优性或过期时间限制不能再加入堆中）将此商品加入方案；反之，若商品利润大于堆中商品利润的最小值，那么可以使用此商品替换堆顶对应的商品。时间复杂度 $O(n\log n)$。

#include <algorithm>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn = 10005;
struct prod
{
    int p, d;
    bool operator<(const prod &b) const { return d < b.d; }
} P[maxn];
int N;

int main()
{
    while (~scanf("%d", &N))
    {
        for (int i = 0; i < N; ++i)
            scanf("%d %d", &P[i].p, &P[i].d);
        sort(P, P + N);
        priority_queue<int, vector<int>, greater<int>> q;
        int res = 0;
        for (int i = 0; i < N; ++i)
            if (P[i].d > (int)q.size())
                res += P[i].p, q.push(P[i].p);
            else
            {
                if (P[i].p > q.top())
                    res += P[i].p - q.top(), q.pop(), q.push(P[i].p);
            }
        printf("%d\n", res);
    }
    return 0;
}

贪心 + 并查集

使用贪心策略，优先考虑卖出利润大的商品，对于每一个商品，在过期之前尽量晚的卖出。

对商品按照利润降序排序，顺序处理。使用并查集维护从过期时间向前数的第一个未占用位置。具体而言，每次考虑商品，查询 $d$ 的根节点 $r$，若根节点非零，则将这个商品在根节点对应的时间卖出，同时将 $r$ 指向 $r-1$，即卖出时间的前一天，此时各节点的根节点代表从当前节点前向数的第一个未占用位置。仅使用路径压缩进行优化，时间复杂度 $O(n\log n)$。

#include <algorithm>
#include <cstdio>
using namespace std;
const int maxn = 10005;
struct prod
{
    int p, d;
    bool operator<(const prod &b) const { return p > b.p; }
} P[maxn];
int N, D, fa[maxn];

int find(int x) { return fa[x] == x ? x : (fa[x] = find(fa[x])); }

int main()
{
    while (~scanf("%d", &N))
    {
        D = 0;
        for (int i = 1; i <= N; ++i)
            scanf("%d%d", &P[i].p, &P[i].d), D = max(D, P[i].d);
        for (int i = 1; i <= D; ++i)
            fa[i] = i;
        sort(P + 1, P + N + 1);
        int res = 0;
        for (int i = 1; i <= N; ++i)
        {
            int x = find(P[i].d);
            if (x)
                res += P[i].p, fa[x] = x - 1;
        }
        printf("%d\n", res);
    }
    return 0;
}