关闭

【Java】设计一个栈,并且push,pop和min三个方法的时间复杂度必须为O(1)

1048人阅读 评论(0) 收藏 举报
分类:

设计一个栈,除pop与push方法,还支持min方法,可返回栈元素中的最小值。

push,pop和min三个方法的时间复杂度必须为O(1)

要保证min的时间复杂度为O(1), 那就不能每次通过遍历整个栈来获取最小值。假设这个栈初始为5:

5; 最小值为5; 压入7,栈变为

7,5;最小值为5; 压入3,栈变为

3,7,5; 最小值为3


这时弹出3,栈变为

7,5; 最小值为5; 弹出7, 栈变为

5; 最小值为5;


注意在7,5这个状态时,最小值都是5. 也就是说,如果我们能标记出栈的每个状态,那么可以直接取出最小值。可以用一个栈来保存最小值。

stack类的peek()方法,查看栈顶对象而不移除它

解法一:

import java.util.Stack;

class NodeWithMin {
	public int value;
	public int min;
	public NodeWithMin(int v, int min) {
		value = v;
		this.min = min;
	}
}

public class StackWithMin extends Stack<NodeWithMin> {
	public void push(int value) {
		int newMin = Math.min(value, min());
		super.push(new NodeWithMin( value, newMin));
	}
	
	public int min() {
		if ( this.isEmpty() ) {
			return Integer.MAX_VALUE;
		}
		else {
			return peek().min;
		}
	}
}


解法二:

import java.util.Stack;

public class StackWithMin2 extends Stack<Integer> {
	//an extra stack to store the minimum in every push operation
	Stack<Integer> s2;
	public StackWithMin2() {
		s2 = new Stack<Integer>();
	}
	
	public void push(int value) {
		if ( value <= min() ) {
			p2.push(value);
		}
		else {
			super.push(value);
		}
	}
	
	public Integer pop() {
		int value = super.pop();
		if( value == min() ) {
			s2.pop();
		}
		return value;
	}
	
	public int min() {
		if (s2.isEmpty()) {
			return Integer.MAX_VALUE;
		}
		else {
			return s2.peek();
		}
	}
}


1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:128524次
    • 积分:3182
    • 等级:
    • 排名:第11349名
    • 原创:180篇
    • 转载:107篇
    • 译文:0篇
    • 评论:10条
    文章分类
    最新评论