《程序员面试金典》集合栈

【 声明：版权所有，转载请标明出处，请勿用于商业用途。 联系信箱：libin493073668@sina.com】





题目链接：http://www.nowcoder.com/practice/69f0ffed01c741c5ae5594a23f7cd739?rp=1&ru=/ta/cracking-the-coding-interview&qru=/ta/cracking-the-coding-interview/question-ranking

题目描述

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

给定一个操作序列int[][2] ope(C++为vector<vector<int>>)，每个操作的第一个数代表操作类型，若为1，则为push操作，后一个数为应push的数字；若为2，则为pop操作，后一个数无意义。请返回一个int[][](C++为vector<vector<int>>)，为完成所有操作后的SetOfStacks，顺序应为从下到上，默认初始的SetOfStacks为空。保证数据合法。

思路

因为这里是使用vector来实现这个栈的，所以相对就容易了，首先我们要注意的是，当辅助栈被清空的时候，我们要注意对于答案vector的处理

class SetOfStacks
{
	public:
		vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size)
		{
			// write code here
			vector<vector<int> > ans;
			int len = ope.size();
			if(len==0 || size<=0)
				return ans;
			vector<int> tmp;
			for(int i = 0; i<len; i++)
			{
				if(ope[i][0]==1)
				{
					if(tmp.size()==size)
					{
						ans.push_back(tmp);
						tmp.clear();
					}
					tmp.push_back(ope[i][1]);
				}
				else
				{
					if(tmp.size()==0)
					{
						int ansSize = ans.size();
						tmp = ans[ansSize-1];
						ans.pop_back();
					}
					tmp.pop_back();
				}
			}
			if(tmp.size())
				ans.push_back(tmp);
			return ans;
		}
};