集合栈之程序员面试经典

题目描述

请实现一种数据结构SetOfStacks，由多个栈组成，其中每个栈的大小为size，当前一个栈填满时，新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。

给定一个操作序列int[][2] ope(C++为vector<vector<int>>)，每个操作的第一个数代表操作类型，若为1，则为push操作，后一个数为应push的数字；若为2，则为pop操作，后一个数无意义。请返回一个int[][](C++为vector<vector<int>>)，为完成所有操作后的SetOfStacks，顺序应为从下到上，默认初始的SetOfStacks为空。保证数据合法。

从题目可以得出A[][0]==1的时候表示push压栈,A[][0]==2的时候表示pop出栈,并且A[][2]的大小为2；

代码如下：

public ArrayList<ArrayList<Integer>> setOfStacks(int[][] ope, int size) {
         ArrayList<ArrayList<Integer>> setofStacks =new ArrayList<ArrayList<Integer>>();
         ArrayList<Integer> inner =new ArrayList<Integer>();
         setofStacks.add(inner);
         for(int i=0;i<ope.length;i++){
        inner =setofStacks.get(setofStacks.size()-1);
        if(ope[i][0]==1){//表示push
        if(inner.size()<size){//先判断该集合栈是否为已经满了
        inner.add(ope[i][1]);
        }else{
        inner =new ArrayList<Integer>();
        inner.add(ope[i][1]);
        setofStacks.add(inner);
        }
        }
        if(ope[i][0]==2){//表示pop出栈
        //首先先判断栈是否为空
        if(setofStacks.get(0).size()==0)
        continue;//因为连最底层的都为空肯定是没有了,所以continue
        //不是最底层的栈为空的话表示就一定至少有两层
        if(inner.size()==0){
        //这里判断inner.size()==0是因为 setofStacks.get(0).size()==0判断底层是否为0;
        //如果为0直接i++;但是如果到达当前的if判断语句，则表示至少有两层
        //这是移除当前空的栈
        setofStacks.remove(setofStacks.size()-1);
        //然后再去得到倒数第二个栈
        inner =setofStacks.get(setofStacks.size()-1);
        }
        //最后进行pop出栈操作
        inner.remove(inner.size()-1);
        }
         }
         return setofStacks;
}