学习了栈和队列的基本知识后,我们要利用这些基本知识实现出更多情况的栈和队列,下面通过一些面试题使我们更灵活的设计和使用栈和队列。
1.利用一个数组实现两个栈
<一>思路:
我们已经学过了栈和数组,数组是一块固定大小的存储空间,栈的栈底是固定的,通过栈顶来push,pop元素。(栈顶top指向栈顶元素)
通过上面的图片,就能很清晰的看到,我们要用一个数组实现两个栈的话,数组的两端是两个栈的栈底,然后向中间靠拢,top1和top2是两个栈的栈顶。
那么什么时候栈满,什么时候栈空呢?
栈满:两个栈共享一块空间,所以栈满是由两个栈一起控制的,当两栈碰面时,栈满,即
top1+1 == top2
栈空:当top1为-1时,栈1为空;当top2为capacity(数组的大小为capacity)时栈2为空。
<二>实现:(c++)
#pragma once
//一个数组实现两个栈
#include <iostream>
#include <cassert>
#include <string.h>
using namespace std;
#define CAPACITY 10
template <class T>
class DouStack
{
public:
DouStack()
: _top1(-1)
, _top2(CAPACITY)
{}
void Push(int nu_stack, T data)
{
assert(nu_stack == 1 || nu_stack == 2);
assert(_top1 + 1 != _top2); //栈满
if (nu_stack == 1)
{
_data[++_top1] = data;
}
else if (nu_stack == 2)
{
_data[--_top2] = data;
}
}
void Pop(int nu_stack)
{
assert(nu_stack == 1 || nu_stack == 2);
if (nu_stack == 1)
{
assert(_top1 != -1); //栈空
_top1--;
}
else if (nu_stack == 2)
{
assert(_top2 != CAPACITY);//栈空
_top2++;
}
}
bool Empty(int nu_stack)
{
assert(nu_stack == 1 || nu_stack == 2);
return (nu_stack == 1) ? (_top1 == -1) : (_top2 == CAPACITY);
}
T& Top(int nu_stack)
{
assert(nu_stack == 1 || nu_stack == 2);
return (nu_stack == 1) ? _data[_top1] : _data[_top2];
}
size_t Size(int nu_stack)
{
assert(nu_stack == 1 || nu_stack == 2);
return (nu_stack == 1) ? (_top1+1) : (CAPACITY - _top2);
}
protected:
T _data[CAPACITY];
int _top1;
int _top2;
};