首页 > 其他 > 详细

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

时间:2014-02-19 22:55:37      阅读:431      评论:0      收藏:0      [点我收藏+]

普通的栈,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;
}


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

原文:http://blog.csdn.net/shoulinjun/article/details/19472985

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!