程序员面试金典 2.7 集合栈

题目

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

给定一个操作序列

int[][2] ope

(C++为

vector<vector<int>>

)，每个操作的第一个数代表操作类型，若为1，则为push操作，后一个数为应push的数字；若为2，则为pop操作，后一个数无意义。请返回一个

int[][]

(C++为

vector<vector<int>>

)，为完成所有操作后的SetOfStacks，顺序应为从下到上，默认初始的SetOfStacks为空。保证数据合法。

我的题解

import java.util.*;

public class SetOfStacks {
public ArrayList<ArrayList<Integer>> setOfStacks(int[][] ope, int size) {
// write code here
int counter = 0;
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> cur = null;
for(int i = 0; i < ope.length; i ++) {
if(ope[i][0] == 1) {
if(counter % size == 0) {
cur = new ArrayList<Integer>();
res.add(cur);
}
cur.add(ope[i][1]);
counter ++;
}else {
cur.remove(cur.size() - 1);
counter --;
if(counter % size == 0) {
res.remove(res.size() - 1);
cur = res.get(res.size() - 1);
}
}
}

return res;
}
}