问题描述
我有这段代码,但我发现以递归方式编写它很困难(二叉搜索树中的插入方法)
public void insert(E element){
BTNode<E> node=root;
boolean stop=false;
while(!stop){
if(element.compareto(node.getElement())<0){
if(node.getLeft()==null){
node.setLeft(new BTNode<E>(element));
stop=true;
} else{
node=node.getLeft();
}
}
else{
if(node.getRight()==null){
node.setRight(new BTNode<E>(element));
stop=true;
} else{
node=node.getRight();
}
}
}
}
解决方法
您可以编写一个方法来接受根和要插入的元素并返回修改后的根。
public BTNode<E> insert(BTNode<E> root,E element) {
if ( root == null ) {
root = new BTNode<E>(element);
} else if ( element.compareTo(root.getElement()) < 0 ) {
root.left = insert(root.left,element);
} else {
root.right = insert(root.right,element);
}
return root;
}