程序员面试金典——3.3集合栈_集合栈程序员面试金典-CSDN博客

本文链接：https://blog.csdn.net/Allenlzcoder/article/details/79877816

程序员面试金典——3.3集合栈

Solution1：主要总结下关于vector不太常用的函数
（1）c.pop_back():删除vector对象c中最后一个元素，若c为空，则函数行为未定义。函数返回void
（2）c.clear():删除c中所有元素。返回void
（3）c.back():返回c中尾元素的引用。若c为空，函数行为未定义。

class SetOfStacks {
public:
    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
        // write code here
        vector<vector<int> > res;
        if(ope.size() == 0 || ope[0].size() == 0)
            return res;
        //用vector模拟栈的操作
        vector<int> vec_stack; //第一个模拟栈，只有当 栈满&&下个操作为push时，才把vec_stack压入带res中
                               //然后清空栈，接着往后遍历操作序列
        for(int i = 0; i < ope.size(); i++) { //遍历所有的操作序列
            if(vec_stack.size() == size && ope[i][0] == 1) { //把栈压入vector的条件
                res.push_back(vec_stack);
                vec_stack.clear();
            }
            if(ope[i][0] == 1) { //push()操作容易
                vec_stack.push_back(ope[i][1]);
            }
            if(ope[i][0] == 2) { //有很多pop()操作则有点麻烦
                if(vec_stack.size() != 0)
                    vec_stack.pop_back();
                else if(vec_stack.size() == 0) {
                    if(res.size() == 0)
                        return res;
                    else { //若当前vec_stack为空，那么就删除res末元素的末元素
                        vec_stack = res.back();
                        vec_stack.pop_back();
                        res.pop_back();
                    }
                }
            }
            if(i == ope.size() - 1 && vec_stack.size() != 0)
                res.push_back(vec_stack);
        }
        return res;
    }
};