堆 02 添加 & 上浮

void add(E e) - 向最大堆中添加元素

  • 从二叉树的角度看,向最大堆中添加元素就是接着在完全二叉树的下一个位置码上元素;
  • 从数组的角度看,就是在数组的最后添加一个元素;
  • 从堆的角度看,新添加的元素不能比它的父元素大,如果比它的父元素大,就涉及到上浮的动作;
// 向堆中添加元素
public void add(E e){
    data.addLast(e);
    siftUp(data.getSize() - 1);
}

void siftUp(int k) - 上浮新添加的元素

  • k是新添加的元素的索引;
  • 若果k > 0,并且其值大于其父节点的值:
    • 新添加的元素和其父元素交换位置;
    • 让游标k追到新添加的元素上浮后的位置;
private void siftUp(int k){
    while(k > 0 && data.get(parent(k)).compareTo(data.get(k)) < 0 ){
        data.swap(k, parent(k));
        k = parent(k);
    }
}

你可能感兴趣的:(堆 02 添加 & 上浮)