面试题 03.03. 堆盘子

题

面试题 03.03. 堆盘子
堆盘子。设想有一堆盘子，堆太高可能会倒下来。因此，在现实生活中，盘子堆到一定高度时，我们就会另外堆一堆盘子。请实现数据结构SetOfStacks，模拟这种行为。SetOfStacks应该由多个栈组成，并且在前一个栈填满时新建一个栈。此外，SetOfStacks.push()和SetOfStacks.pop()应该与普通栈的操作方法相同（也就是说，pop()返回的值，应该跟只有一个栈时的情况一样）。 进阶：实现一个popAt(int index)方法，根据指定的子栈，执行pop操作。

当某个栈为空时，应当删除该栈。当栈中没有元素或不存在该栈时，pop，popAt 应返回 -1.

示例1:

输入：
[“StackOfPlates”, “push”, “push”, “popAt”, “pop”, “pop”]
[[1], [1], [2], [1], [], []]
输出：
[null, null, null, 2, 1, -1]
示例2:

输入：
[“StackOfPlates”, “push”, “push”, “push”, “popAt”, “popAt”, “popAt”]
[[2], [1], [2], [3], [0], [0], [0]]
输出：
[null, null, null, null, 2, 1, 3]

代码

class StackOfPlates {
	struct Dish{
		stack<int> s;
		int size = 0;
		Dish() :size(0),s(stack<int>()) {}
	};
	list<Dish*> plaste;
	int capacity = 0;


public:
	StackOfPlates(int cap) {
		capacity = cap;
	}

	void addS() {
		plaste.push_back(new Dish());
	}

	void push(int val) {
		if (!capacity)
		{
			return;
		}
		if (plaste.size()==0||plaste.back()->size >= capacity)
		{
			addS();
		}

		plaste.back()->s.push(val);
		plaste.back()->size++;
	}

	int pop() {
		if (plaste.size()<=0)
		{
			return -1;
		}
		int result = plaste.back()->s.top();
		plaste.back()->s.pop();
		plaste.back()->size--;
		if (plaste.back()->size==0)
		{
			plaste.pop_back();
		}

		return result;
	}

	int popAt(int index) {
		if (plaste.size() <= 0)
		{
			return -1;
		}
		for (auto ir = plaste.begin(); ir != plaste.end(); ir++, index--) {
			auto data=*ir;
			if (index==0)
			{
				data->size--;
				if (data->size==0)
				{
					plaste.erase(ir);
				}
				int result = data->s.top();
				data->s.pop();
				return result;
			}
		}

		return -1;
	}
};