简单来说,就是在已经有的数据结构上加上一些新的操作而已。
比如:为了快速找到从小到大顺序是i的数值。可以在红黑树的基础上加上size关键字。
在进行插入和删除的操作时,顺便更新size,在查找时也会很快就能找到。
以下是查找的代码:
public Node get(Node node, int i) {
if (node.left.size == i) {
return node;
}
if (i < node.left.size) {
return this.get(node.left, i);
}
return this.get(node.right, i-node.left.size-1);
}
在进行添加时候,只要依次将父节点及父节点以上的节点进行重计算处理就行
public int getSize() {
return this.left.size+this.right.size+1;
}
public void setSize() {
this.size = this.getSize();
if(this.parent!=null) {
this.parent.setSize();
}
}
旋转时候,只需要将旋转节点及旋转后的父节点进行重计算size就行
node.setSize();
node.parent.setSize();
删除时候,只要把删除节点的父节点进行重计算size就行
经检验,代码正确。