【每日一题】餐盘栈

1172. 餐盘栈

关键词：栈、模拟、优先队列

题目来源：1172. 餐盘栈 - 力扣（Leetcode）

题目描述

 T栈
 T模拟
 T优先队列

我们把无限数量 ∞ 的栈排成一行，按从左到右的次序从 0 开始编号。每个栈的的最大容量 capacity 都相同。

实现一个叫「餐盘」的类 DinnerPlates：

• DinnerPlates(int capacity) - 给出栈的最大容量 capacity。
• void push(int val) - 将给出的正整数 val 推入 从左往右第一个 没有满的栈。
• int pop() - 返回 从右往左第一个 非空栈顶部的值，并将其从栈中删除；如果所有的栈都是空的，请返回 -1。
• int popAtStack(int index) - 返回编号 index 的栈顶部的值，并将其从栈中删除；如果编号 index 的栈是空的，请返回 -1。

数据范围
1 <= capacity <= 20000
1 <= val <= 20000
0 <= index <= 100000
最多会对 push，pop，和 popAtStack 进行 200000 次调用。

问题分析

考虑使用数组stk来模拟栈，设编号为idx的栈的栈顶元素的下标为top，则在数组中的下标为idx*capacity+top，每个栈的栈顶位置可使用数组来维护，空位可采用一个优先队列来维护。

对于push，若有空位，则找到最小的位置，否则新开一个栈

对于pop，返回stk末尾元素即可，注意，是末尾元素而不是末尾位置

对于popAtStack，找到对应栈并让栈顶元素出栈即可

代码实现

class DinnerPlates {
private:
    int capacity;
    vector<int> stk;
    vector<int> top;
    set<int> freePos;
public:
    DinnerPlates(int capacity) {
        this->capacity = capacity;
    }
    void push(int val) {
        // 没有空位则新开一个栈
        if (freePos.empty()) {
            int pos = stk.size();
            stk.emplace_back(val);
            if (pos % capacity == 0) {
                top.emplace_back(0);
            } else {
                top.back()++;
            }
        }
        // 有空位则去除最小位置
        else {
            int pos = *freePos.begin();
            freePos.erase(pos);
            stk[pos] = val;
            int index = pos / capacity;
            top[index]++;
        }
    }

    int pop() {
        // 删除末尾空栈
        while (!stk.empty() && freePos.count(stk.size() - 1)) {
            stk.pop_back();
            int pos = *freePos.rbegin();
            freePos.erase(pos);
            if (pos % capacity == 0) {
                top.pop_back();
            }
        }
        // 全部栈被删除
        if (stk.empty()) {
            return -1;
        }
        // 末尾非空栈
        else {
            int pos = stk.size() - 1;
            int val = stk.back();
            stk.pop_back();
            if (pos % capacity == 0) {
                top.pop_back();
            } else {
                top.back() = top.size() - 2;
            }
            return val;
        }
    }

    int popAtStack(int index) {
        // 末尾空栈
        if (index >= top.size()) {
            return -1;
        }
        // 前部空栈
        int stackTop = top[index];
        if (stackTop < 0) {
            return -1;
        }
        // 非空栈则删除栈顶元素
        top[index]--;
        int pos = index * capacity + stackTop;
        freePos.emplace(pos);
        return stk[pos];
    }
};