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

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

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

题目分析:

这道题目基本没什么技巧,就是堆栈入栈的过程,相当于用多个数组来实现一个堆栈,如果一个数组满了,则使用另一数组,如果当且数组空了,则把该数组删除,从它前一个数组中操作,实现的时候注意数组交换数组时应该如何判断。

题目代码实现:

#include<iostream>
#include<vector>
using namespace std;
class SetOfStacks {
public:
    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
        // write code here
		int sz=ope.size();
		if(sz==0)
			return ope;
		vector<vector<int> > vecvec;
		vector<int> vec;
		int top=0;
		for(int i=0;i<sz;i++)
		{
			if(ope[i][0]==1)
			{
				vec.push_back(ope[i][1]);
				if(++top==size)
				{
					vecvec.push_back(vec);
					vec.clear();
					top=0;
				}
			}
			else
			{
				if(top==0)
				{
					vec=vecvec.back();
					vecvec.pop_back();
					top=size;
				}
				vec.pop_back();
				top--;
			}
		}
		if(!vec.empty())
			vecvec.push_back(vec);
		return vecvec;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值