JD3.3栈—集合栈

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

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

import java.util.*;
/*
* 思路：先定义curArray代表当前栈，加入集合中
* 再先判断二维数组有多少行，代表操作次数，判断第0列是1还是2，从而判断是push还是pop
* */
public class SetOfStacks {
    public ArrayList<ArrayList<Integer>> setOfStacks(int[][] ope, int size) {

        ArrayList<ArrayList<Integer>>list=new ArrayList<>();

        //定义第一个栈
        ArrayList<Integer>curArray=new ArrayList<>(size);

        //将第一个栈加入到集合
        list.add(curArray);

        for(int i=0;i<ope.length;i++)
        {
            //switch语句判断第0列是1还是2（也可if语句判断）
            switch(ope[i][0])
            {
                //1:push
                case 1:
                    //若当前数组未满，数据值入当前栈
                    if(curArray.size()!=size){
                        curArray.add(ope[i][1]);
                    }
                    //若当前栈已满，则新建一个栈，并将栈加入集合，并将数据加入到新建栈中
                    else{
                        curArray=new ArrayList<Integer>(size);
                        list.add(curArray);
                        curArray.add(ope[i][1]);
                    }
                    break;
                //2:pop
                case 2:
                    //若当前栈不为空，则直接删除当前栈的最后一个元素
                    if(curArray.size()!=0){
                        curArray.remove(curArray.size()-1);
                    }
                    //若当前栈为空，则说明栈集合中最后一个栈里面元素被pop空了，
                    else{
                        list.remove(list.size()-1);//则把最后一个空栈删除，
                        curArray=list.get(list.size()-1);//curArray指向栈集合中新的最后一个栈作为当前栈
                        curArray.remove(curArray.size()-1);//当前栈执行pop操作，将当前栈的栈顶元素删除
                    }
                    break;
            }
        }
        return list;
    }
}