feat(min heap): Add method of getting the first element

agribeau · agribeau · commit d0a9d74f7422 · 2018-10-15T11:26:28.000+08:00
diff --git a/src/data_structure/heap/MinHeap.java b/src/data_structure/heap/MinHeap.java
@@ -88,6 +88,24 @@ public boolean remove(Object o) {
         }
     }
 
+    /**
+     * 移除最小堆的首节点
+     *
+     * @return 首节点元素
+     */
+    public E poll() {
+        int s = data.size();
+        if (s == 0) {
+            return null;
+        }
+
+        E result = data.get(0);
+        E x = data.remove(s - 1);
+        siftDown(0, x);
+
+        return result;
+    }
+
     /*=============================================================*/
     /*========================== 私有方法 ==========================*/
     /*=============================================================*/
