数据结构---设计一个栈,push, pop, min 时间复杂度都是 O(1)

普通的栈,push, pop 操作的复杂度是 O(1), 但是如果要找出其中的最小值,则需要 O(N)的时间。题目要求 min 复杂度也是 O(1), 做法便是 空间换时间,每一步栈的最小值都用一个数组记录下来。。// copyright @ L.J.SHOU Feb.17, 2014// design a stack// min, push, pop take O(1)
摘要由CSDN通过智能技术生成

普通的栈,push, pop 操作的复杂度是 O(1), 但是如果要找出其中的最小值,则需要 O(N)的时间。

题目要求 min 复杂度也是 O(1), 做法便是 空间换时间,每一步栈的最小值都用一个数组记录下来。。


// copyright @ L.J.SHOU Feb.17, 2014
// design a stack
// min, push, pop take O(1) time
#include <iostream>
using namespace std;

/* 无穷大 */
const int INT = 0xfffff;

template<typename T>
class MyStack{
public:
  MyStack(int capacity=100)
    :capacity_(capacity), top_(-1){
	  stack_ = new T[capacity_]; 
	  min_   = new T[capacity_]; 
	}

  ~MyStack(){ 
    delete [] stack_; 
    delete [] min_; 
  }

  void push(const T& t){
    if(top_ < capacity_){
	  stack_[top_ + 1] = t;

	  if(top_ < 0 || min_[top_] > t){
	    min_[top_ + 1] = t;
	  }
	  else{
	    min_[top_ + 1] = min_[top_];
	  }

	  ++ top_;
	}
	else{
	  cerr << "stack is full" << endl;
	}
  }

  void pop(void){
    if(top_ >= 0){
	  -- top_;
	}
	else{
	  cerr << "stack is empty" << endl;
	}
  }

  T top(){
    if(top_ >= 0){
	  return stack_[top_];
	}
	else{
	  cerr << "stack is empty" << endl;
	  return INT;
	}
  }

  T min(){
    if(top_< 0){ 
	  cerr << "stack is empty" << endl;
	  return INT;
	}
    return min_[top_];
  }

private:
  int capacity_; //max_size
  int top_;    
  T *stack_;
  T *min_;
};

int main(void)
{
  MyStack<int> s;

  s.push(1);
  s.push(2);
  s.push(3);

  cout << s.top() << endl;
  s.pop();
  cout << s.top() << endl;
  cout << s.min() << endl;
  s.pop();
  cout << s.top() << endl;
  s.pop();
  cout << s.top() << endl;

  return 0;
}


  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值