【IT笔试面试题整理】包含min函数的栈

【试题描述】定义的数据结构,请在类型中实现一个能够得到栈的最小元素的函数。 在该栈中,调用min、push、pop的时间复杂度都是O(1)。

【试题来源】未知

【参考代码】

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#include <iostream>
#include <stack>
using namespace std;

template<typename T>
class StackWithMin {

private:
stack<T> dataStack;
stack<T> minStack;

public:
void push(const T &element);
T pop();
T min();
};

template<typename T>
void StackWithMin<T>::push(const T &element) {

dataStack.push(element);

if(minStack.empty()) {
minStack.push(element);
} else if(element < minStack.top()) {
minStack.push(element);
} else {
minStack.push(minStack.top());
}
}

template<typename T>
T StackWithMin<T>::pop() {

if(dataStack.empty()) {
throw "Stack is empty.";
}

T topValue = dataStack.top();
dataStack.pop();
minStack.pop();

return topValue;
}

template<typename T>
T StackWithMin<T>::min() {

if(dataStack.empty()) {
throw "Stack is empty.";
}

return minStack.top();
}


int main() {
StackWithMin<int> stackWithMin;
stackWithMin.push(3);
stackWithMin.push(4);
stackWithMin.push(2);
stackWithMin.push(6);
cout << stackWithMin.min() << endl;
stackWithMin.pop();
stackWithMin.pop();
cout << stackWithMin.min() << endl;
return 0;
}
坚持原创技术分享,您的支持将鼓励我继续创作!