设计包含min函数的栈O（1）时间复杂度（笔试）

最新推荐文章于 2025-05-29 16:16:32 发布

原创最新推荐文章于 2025-05-29 16:16:32 发布 · 533 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#数据结构 #system

收录于

Algorithm/Data structure 专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种在栈中实现min函数的方法，该方法可以在O(1)时间内获取栈内最小元素，通过使用额外数组dp来记录每个位置的最小值。

设计包含min函数的栈。定义栈的数据结构，要求添加一个min函数，能够得到栈的最小元素。要求函数min、push以及pop的时间复杂度都是O(1)。

在入栈的时候就用一个数组dp对当前栈内的元素最小值保存

#include <iostream>
#include <cstdio>
#include <string>
using namespace std;
class STACK {
private:
	int size;
	int cap;
	int buf[100];
	int dp[100];
	int top;
public:
	STACK():size(0),cap(1000), top(1) {
		dp[0] = INT_MAX;
	}
	bool empty() {return top == 0;}
	void push(const int a) {
		buf[top] = a;
		dp[top] = dp[top-1]>a? a:dp[top-1];
		top++;
	}
	int pop() {
		top--;
		int temp = buf[top];
		return temp;
	}
	int min()const {
		return dp[top-1];
	}
	/*void prt() {
		for(int i= 0; i< top;i++) {
			cout<<dp[i]<<" ";
		}
		cout<<endl;
	}*/
};
int main() {
	STACK mystack;
	int n, a;
	cin>>n;
	for(int i = 0; i<n; i++) {
		cin >>a;
		mystack.push(a);
	}

	cout<<"The min is  "<<mystack.min()<<endl;
	cout<<"Top is  "<<mystack.pop()<<endl;
//	mystack.prt();
	system("pause");
}