题目描述
请实现一种数据结构SetOfStacks,由多个栈组成,其中每个栈的大小为size,当前一个栈填满时,新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。
给定一个操作序列int[][2] ope(C++为vector<vector<int>>),每个操作的第一个数代表操作类型,若为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;
}
}