双栈使堆栈的min操作时间复杂度为O(1)

要求:设计一个栈的数据结构,含有min()操作来返回当前栈的最小值,要求min(),push

,pop的时间复杂度均为O(1).

主栈记录元素,使用一个辅助栈来记录最小值。

1.入栈:只要压入的值小于辅助栈栈顶元素,则同时将其压入辅助栈。

2.出栈:当主栈栈顶元素与辅助栈栈顶元素相等时,这说明最小元素出栈,所以辅助栈也随之出栈,其他情况仅对主栈操作就行

3.min:直接获得辅助栈栈顶元素,即当前主栈中的最小元素。


另一种实现方式是:采用结构体,每次压入元素及最小值,此时的空间复杂度高于上一种实现方式。

代码:

#include<stdio.h>
#define Slen 100
#define Elen 10
int stack[Slen];
int mstack[Slen];
int top = 0, mtop = 0;
void push(int a) {
	stack[top++] = a;
	if (mtop == 0) mstack[mtop++] = a;
	else if (a < mstack[mtop-1]) mstack[mtop++] = a;
}
int pop() {
	if (stack[top - 1] == mstack[mtop - 1])
		mtop--;
	return stack[--top];
}
int min() {
	return mstack[mtop - 1];
}
int main(void)
{
	int a[10] = { 4,3,2,1,2,3,5,6,3,9 };
	for (int i = 0; i < 10; i++)
		push(a[i]);
	for (int i = 0; i < 7; i++)
	{
		int b = pop();
		printf("%d ", b);
	}
	printf("\n%d", min());
}

猜你喜欢

转载自blog.csdn.net/Titanium_S/article/details/79766046
今日推荐