代码改变世界

Min Stack

2015-04-07 13:08  笨笨的老兔子  阅读(166)  评论(0)    收藏  举报

设计一个最小堆，要求实现push、pop、top、getMin几个功能。

思路：主要问题在于getMin，必须是一个常数级的查找返回，因此最好是每push一个就和当前min进行比较，始终保存min。

  1. class MinStack {
  2. public:
  4.     MinStack()
  5.     {
  6.         head = new ListNode(INT_MAX);
  7.     }
  8.     void push(int x) {
  9.         ListNode* tmpNode=new ListNode(x);
  10.         if (x < head->val)
  11.         {
  12.             head->val = x;
  13.         }
  14.         tmpNode->next = head->next;
  15.         head->next = tmpNode;
  16.     }
  18.     void pop() {
  19.         if (head->next )
  20.         {
  21.             if (head->val == head->next->val)
  22.             {
  23.                 ListNode* tmp = head->next->next;
  24.                 head->val = INT_MAX;
  25.                 while (tmp)
  26.                 {
  27.                     if (tmp->val < head->val)
  28.                         head->val = tmp->val;
  29.                     tmp = tmp->next;
  30.                 }
  31.             }
  32.             head->next = head->next->next;
  33.         }
  34.     }
  36.     int top() {
  37.         if (head->next)
  38.             return head->next->val;
  39.     }
  41.     int getMin() {
  42.         if (head->next)
  43.             return head->val;
  44.     }
  45. private:
  46.     struct ListNode {
  47.         int val;
  48.         ListNode *next;
  49.         ListNode(int x) : val(x), next(NULL) {}
  50.     };
  51.     ListNode* head;
  52. };