nil = {
var nil;
class AANode {
constructor (key, level, left, right) {
[this.key, this.left, this.right, this.level] = [key,left,right,level]
}
find (key) {
if (this == nil) return null;
if (this.key == key) return this;
if (this.key < key) return this.right.find(key);
return this.left.find(key)
}
clone () {
if (this == nil) return nil;
return new AANode (this.key, this.level, this.left.clone(), this.right.clone())
}
skew () {
if (this.left.level == this.level) {
let q = this.left;
this.left = q.right;
q.right = this;
return q;
}
else return this;
}
// Rearranges a black node with two red nodes in cascade to the right.
// Returns the new root of this subtree
split () {
// Note: must not split the nil node
if (this != nil && this.right.right.level == this.level) {
let q = this.right;
this.right = q.left;
q.left = this;
q.level += 1;
return q;
}
else return this
}
// Inserts key into the tree rooted at this node.
// Returns the inserted node
insert (key) {
let p = this;
if (p == nil)
p = new AANode(key, 1, nil, nil);
else if (key < p.key)
p.left = p.left.insert(key);
else if (key > p.key)
p.right = p.right.insert(key);
else
throw "Duplicate Key: "+key;
return p.skew().split();
}
// Makes sure the level of this node respects the AA level invariant,
// i.e., that is, 1 plus the level of its black children
updateLevel() {
let idealLevel = 1 + Math.min(this.left.level, this.right.level);
if (this.level > idealLevel) {
this.level = idealLevel;
if (this.right.level > idealLevel)
this.right.level = idealLevel;
}
}
// Fixes up the tree after a node is deleted making sure it still
// obeys the AA tree invariant by performing appropriate skews and splits.
// Returns the updated node
fixupAfterDelete() {
this.updateLevel();
let p = this.skew();
p.right = p.right.skew();
p.right.right = p.right.right.skew();
p = p.split();
p.right = p.right.split();
return p
}
// Returns the descendant node containing the immediately smaller key
leftReplacement () {
let p = this.left;
while (p.right != nil) p = p.right;
return p;
}
// Returns the descendant node containing the immediately bigger key
rightReplacement () {
let p = this.right;
while (p.left != nil) p = p.left;
return p;
}
// Returns this node after deleting key from the tree rooted at it
delete (key) {
if (this == nil) throw "not found: "+key;
if (key < this.key)
this.left = this.left.delete(key);
else if (key > this.key)
this.right = this.right.delete(key);
else {
if (this.left == nil && this.right == nil) // leaf node?
return nil; // just unlink the node
else if (this.left == nil) { // no left child?
let r = this.rightReplacement(); // get replacement from right
this.key = r.key; // copy replacement contents here
this.right = this.right.delete(r.key);// delete replacement
}
else { // no right child?
let r = this.leftReplacement(); // get replacement from left
this.key = r.key; // copy replacement contents here
this.left = this.left.delete(r.key); // delete replacement
}
}
return this.fixupAfterDelete(); // fix structure after deletion
}
}
nil = new AANode ('nil',0);
nil.left = nil.right = nil;
return nil
}