题目描述
请实现一种数据结构SetOfStacks,由多个栈组成,其中每个栈的大小为size,当前一个栈填满时,新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。
给定一个操作序列int[][2] ope(C++为vector<vector<int>>),每个操作的第一个数代表操作类型,若为1,则为push操作,后一个数为应push的数字;若为2,则为pop操作,后一个数无意义。请返回一个int[][](C++为vector<vector<int>>),为完成所有操作后的SetOfStacks,顺序应为从下到上,默认初始的SetOfStacks为空。保证数据合法。
思路:用arraylist来实现栈,放进外层的arraylist里面。这个要用一个很重要的特点,就是在ans.add(temp),之后,temp.add(1),这样ans里面的temp还是会变化,因为ans里面传入的是引用不是值,所以在当前栈没有空或者没有满之前,上下文中的栈都可以使用这个引用,当发现这个stack空了,或者满了,这个时候就应该再new一个stack加到ans里面,或者去掉一个stack,还是用相同的名字来指向新的stack,进行下一补操作。
import java.util.*; public class SetOfStacks { public ArrayList<ArrayList<Integer>> setOfStacks(int[][] ope, int size) { // write code here //要用arraylist来实现stack ArrayList<ArrayList<Integer>> ans=new ArrayList<ArrayList<Integer>>(); ArrayList<Integer> temp=new ArrayList<Integer>(); ans.add(temp); for(int i=0;i<ope.length;i++){ if(ope[i][0]==1){//push if(temp.size()<size){ temp.add(ope[i][1]); }else{ temp=new ArrayList<Integer>(); temp.add(ope[i][1]); ans.add(temp); } }else{ if(temp.size()>0){ temp.remove(temp.size()-1); }else{ ans.remove(ans.size()-1); temp=ans.get(ans.size()-1); temp.remove(temp.size()-1); } } } return ans; } }