您的位置:首页 > 其它

集合栈

2016-05-05 18:59 337 查看
对入栈和出栈操作,top都需要相应加减,对一些条件的判断。

/*****************************************************
* \file SetOfStacks.cpp
* \date 2016/05/05 17:49

* \问题描述:
请实现一种数据结构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>
using namespace std;
#include <vector>
class SetOfStacks {
public:
vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
// write code here
if (ope.size()==0)
{
return ope;
}
vector<vector<int>> vec_result;
vector<int> vec;
int top = 0;
for (int i = 0; i < ope.size();i++)
{
if (ope[i][0]==1)   //按顺序执行,则不需要if--else
{
vec.push_back(ope[i][1]);
if (++top==size)  //栈满,则新建一个栈
{
vec_result.push_back(vec);
vec.clear();
top = 0;
}
}
else
{
if (top==0)  //栈为空,则从上一个栈出
{
vec = vec_result.back();
vec_result.pop_back();
top = size;
}
vec.pop_back();
top--;
}
}

//vector<vector<int> > st;
//vector<int> temp;
//for (int i = 0; i < ope.size(); i++){
//    if (ope[i][0] == 1){//入栈
//        if (temp.size() == size){//栈满,则新建一个栈,再入栈
//            st.push_back(temp);
//            temp.clear();
//            temp.push_back(ope[i][1]);
//        }
//        else
//            temp.push_back(ope[i][1]);
//    }
//    if (ope[i][0] == 2){//出栈
//        if (temp.size() != 0)//当前栈不为空,直接出栈
//            temp.pop_back();
//        else if (st.size() != 0){//当前栈为空,则从上一个栈出
//            temp = st[st.size() - 1];
//            temp.pop_back();
//            st.pop_back();
//        }
//    }
//}

if (!vec.empty())   //对于没有满size但又有元素的vec,最后压入返回结果
{
vec_result.push_back(vec);
}
return vec_result;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: