算法导论之第十四章-数据结构的扩张

简单来说,就是在已经有的数据结构上加上一些新的操作而已。
比如:为了快速找到从小到大顺序是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就行
经检验,代码正确。

猜你喜欢

转载自blog.csdn.net/qq_33321609/article/details/87005165