面试题 03.03. 堆盘子

堆盘子。设想有一堆盘子，堆太高可能会倒下来。因此，在现实生活中，盘子堆到一定高度时，我们就会另外堆一堆盘子。请实现数据结构SetOfStacks，模拟这种行为。SetOfStacks应该由多个栈组成，并且在前一个栈填满时新建一个栈。此外，SetOfStacks.push()和SetOfStacks.pop()应该与普通栈的操作方法相同（也就是说，pop()返回的值，应该跟只有一个栈时的情况一样）。 进阶：实现一个popAt(int index)方法，根据指定的子栈，执行pop操作。

当某个栈为空时，应当删除该栈。当栈中没有元素或不存在该栈时，pop，popAt 应返回 -1.

示例1:

 输入：
["StackOfPlates", "push", "push", "popAt", "pop", "pop"]
[[1], [1], [2], [1], [], []]
 输出：
[null, null, null, 2, 1, -1]
示例2:

 输入：
["StackOfPlates", "push", "push", "push", "popAt", "popAt", "popAt"]
[[2], [1], [2], [3], [0], [0], [0]]
 输出：
[null, null, null, null, 2, 1, 3]

class StackOfPlates {
    int n = 0;//每一个栈可以放多少个盘子
    vector<vector<int> *> stk;//使用数组储存栈数组指针
    vector <int>idx;//储存对应栈数组顶部元素的下标
public:
    StackOfPlates(int cap) {
        n = cap;//每一个栈可以放多少个盘子
    }
    
    void push(int val) {
        if(n == 0){ 
            return ;
        }
        //栈空或者一个栈放满新建栈数组
        if(stk.empty() || idx.back() == n){
            stk.push_back(new vector<int>(n+1));
            idx.push_back(0);//push_back() 在Vector最后添加一个元素（参数为要插入的值） 
        }
        auto &topstk = *stk.back();
        topstk[++idx.back()] = val;//.back();返回的的是最后一个元素的引用。     
    }
    
    int pop() {
        if(stk.empty()){
            return -1;
        }
        auto &topstk = *stk.back();
        int res = topstk[idx.back()--];
        if(idx.back() == 0){//栈中没有元素，删掉栈
            stk.pop_back();
            idx.pop_back();
        }
        return res;
    }
    
    int popAt(int index) {
        if(stk.size() <= index){
            return -1;
        }
        auto &idxstk = *stk[index];
        int res = idxstk[idx[index]--];
        if(idx[index] == 0){
            stk.erase(stk.begin()+index);
            idx.erase(idx.begin()+index);
        }
        return res;

    }
};
 

/**
 * Your StackOfPlates object will be instantiated and called as such:
 * StackOfPlates* obj = new StackOfPlates(cap);
 * obj->push(val);
 * int param_2 = obj->pop();
 * int param_3 = obj->popAt(index);
 */