用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
一开始考虑的时候:两个栈,肯定一个用作入队,一个用作出队,于是会有下面这样的代码:
class CQueue {
private Stack<Integer> s1;
private Stack<Integer> s2;
public CQueue() {
s1 = new Stack<>();
s2 = new Stack<>();
}
public void appendTail(int value) {
s1.add(value);
}
public int deleteHead() {
if (s1.isEmpty()) {
return -1;
}
while (!s1.isEmpty()) {
s2.add(s1.peek());
s1.pop();
}
int rec = s2.peek();
s2.pop();
while (!s2.isEmpty()) {
s1.add(s2.peek());
s2.pop();
}
return rec;
}
}
emmm,时间看似不太行… 有一个很明显的问题:如果是连续出队的话,两个栈之间元素的复制步骤将有很大一部分是重复的。事实上,如果以stack1为入队用,stack2为出队用,那么stack2不为空的时候,直接返回队头元素,这样的做法才是高效的。
class CQueue {
private Stack<Integer> s1;
private Stack<Integer> s2;
public CQueue() {
s1 = new Stack<>();
s2 = new Stack<>();
}
public void appendTail(int value) {
s1.add(value);
}
public int deleteHead() {
if (s2.isEmpty()) {
while (!s1.isEmpty()) {
s2.add(s1.peek());
s1.pop();
}
}
// s2还是空的,那么说明队列中没有任何元素
if (s2.isEmpty()) {
return -1;
}
int rec = s2.peek();
s2.pop();
return rec;
}
}
这样子就差不多了。
这道题在《剑指Offer》中,本意还是要我们用模板实现,这里我们简单复习一下C++和Java的模板写法。
函数模板示例 (C++)
#include <iostream>
using namespace std;
template<typename T>
inline T const& Max(T const& a, T const& b) {
return a < b ? b : a;
}
int main() {
cout << Max(1, 2) << endl;
cout << Max('a', 'b') << endl;
getchar();
return 0;
}
类模板示例 (C++)
#include <iostream>
#include <vector>
#include <cstdlib>
#include <string>
#include <stdexcept>
using namespace std;
template <class T>
class Stack {
private:
vector<T> elems; // 元素
public:
void push(T const&); // 入栈
void pop(); // 出栈
T top() const; // 返回栈顶元素
bool empty() const{ // 如果为空则返回真。
return elems.empty();
}
};
template <class T>
void Stack<T>::push (T const& elem)
{
// 追加传入元素的副本
elems.push_back(elem);
}
template <class T>
void Stack<T>::pop ()
{
if (elems.empty()) {
throw out_of_range("Stack<>::pop(): empty stack");
}
// 删除最后一个元素
elems.pop_back();
}
template <class T>
T Stack<T>::top () const
{
if (elems.empty()) {
throw out_of_range("Stack<>::top(): empty stack");
}
// 返回最后一个元素的副本
return elems.back();
}
int main()
{
try {
Stack<int> intStack; // int 类型的栈
Stack<string> stringStack; // string 类型的栈
// 操作 int 类型的栈
intStack.push(7);
cout << intStack.top() <<endl;
// 操作 string 类型的栈
stringStack.push("hello");
cout << stringStack.top() << std::endl;
stringStack.pop();
stringStack.pop();
getchar();
}
catch (exception const& ex) {
cerr << "Exception: " << ex.what() <<endl;
getchar();
return -1;
}
}
模板类和模板方法 (Java)
class vectorElement<T>{
public vectorElement<T> next=null;
public vectorElement<T> previous=null;
public T datum=null;
public vectorElement(T datum){ this.datum=datum; }
}