集合栈

题目描述

请实现一种数据结构SetOfStacks，由多个栈组成，其中每个栈的大小为size，当前一个栈填满时，新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。给定一个操作序列int[][2] ope(C++为vector<vector<int>>)，每个操作的第一个数代表操作类型，若为1，则为push操作，后一个数为应push的数字；若为2，则为pop操作，后一个数无意义。请返回一个int[][](C++为vector<vector<int>>)，为完成所有操作后的SetOfStacks，顺序应为从下到上，默认初始的SetOfStacks为空。保证数据合法。

**1.添加元素时，确认stacklist的当前活动栈是否满**  

> 满 ， 开辟新栈  

> 不满，放入元素

**2.弹出元素时，确认stacklist当前活动栈是否为空**  

>空，则看当前栈是否为第一个栈，若是，则无法弹出，打印empty；若否，则删除当前栈，活动栈仍为栈的当前大小  

>不空，则弹出最后的元素

import java.util.*;

public class SetOfStacks {
    public ArrayList<ArrayList<Integer>> setOfStacks(int[][] ope, int size) {
        // write code here        
        
        ArrayList<ArrayList<Integer>> stacklist = new ArrayList<>();
        
        int length = ope.length;
        ArrayList<Integer> cur = new ArrayList<>();
        
        stacklist.add(cur);
        for (int i = 0; i < length; i++) {
            if(ope[i][0]==1)//入栈
            {
                if(cur.size()==size)
                {
                    cur = new ArrayList<>();
                    stacklist.add(cur);
                }
                cur.add(ope[i][1]);
            }
            else if(ope[i][0]==2)//出栈
            {
                if(cur.size()==0)
                {
                    stacklist.remove(stacklist.size()-1);
                    cur = stacklist.get(stacklist.size()-1);
                }
                cur.remove(cur.size()-1);
            }
        }
        return stacklist;
    }
}