private AANode<T> insert(T element, AANode<T> node) {
if (node == null) {
node = new AANode<>(element);
} else if (element.compareTo(node.getValue()) < 0) {
node.setLeft(insert(element, node.getLeft()));
} else if (element.compareTo(node.getValue()) > 0) {
node.setRight(insert(element, node.getRight()));
} else {
return node;
}
node = skew(node);
node = split(node);
return node;
}
@Override
public T next() {
if (!hasNext()) {
throw new NoSuchElementException();
}
AANode<T> current = stack.pop();
pushLeftChildren(current.getRight());
return current.getValue();
}
private AANode<T> remove(T element, AANode<T> node) {
if (node == null) {
return node;
} else if (element.compareTo(node.getValue()) > 0) {
node.setRight(remove(element, node.getRight()));
} else if (element.compareTo(node.getValue()) < 0) {
node.setLeft(remove(element, node.getLeft()));
} else {
if (node.getLevel() == 1) {
return null;
} else if (node.getLeft() == null) {
AANode<T> left = getSuccessor(node);
node.setRight(remove(left.getValue(), node.getRight()));
node.setValue(left.getValue());
} else {
AANode<T> left = getPredecessor(node);
node.setLeft(remove(left.getValue(), node.getLeft()));
node.setValue(left.getValue());
}
}
node = decreaseLevel(node);
node = skew(node);
node.setRight(skew(node.getRight()));
if (node.getRight() != null) {
node.getRight().setRight(skew(node.getRight().getRight()));
}
node = split(node);
node.setRight(split(node).getRight());
return node;
}
