二叉树节点的存储结构
首先定义二叉树结构的节点存储结构:
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
插入节点
过程分析:
- 每次插入节点需要明确知道被插入的父节点以及被插入节点的位置(左右)。
- 如果为空树,则直接当前节点作为根节点。
- 如果不为空树:
如果新节点值比当前节点值大,则往当前节点的右子树中找位置。
如果新节点值比当前节点值小,则往当前节点的左子树中找位置。
循环在子树中找,如果当前节点为空即null,说明该左子树或右子树已经达到了叶子节点,则将新节点插入到该叶子节点位置(原来的叶子节点作为新节点的父节点)。
//插入节点一定是在叶子节点处插入
public void insertNode(int val) {
//封装数据,构造二叉树节点
TreeNode newNode = new TreeNode(val);
//如果当前二叉树是一个空树,则把新节点放在root根节点位置
if (rootNode == null) {
rootNode = newNode;
return;
}
TreeNode currentNode = rootNode;
TreeNode parentNode = rootNode;
//循环找到新节点值对应节点的位置
while (true) {
parentNode = currentNode;
//如果新节点值小于当前节点值,说明在当前节点的左子树中
if (currentNode.val > val) {
currentNode = currentNode.left;
if (currentNode == null) {
parentNode.left = newNode;
return;
}
}else {
//如果大节点值大于当前节点值,说明在当左节点的右子树中
currentNode = currentNode.right;
if (currentNode == null) {
parentNode.right = newNode;
return;
}
}
}
}
按值查找节点
按值查找相对简单一些,更综合的查找还有二叉树的各种遍历算法。
由于二叉树相对有序,首先判断当前节点是否是要查找值的节点,如果是直接返回,否则:
如果当前节点值大于目标值(要查找的值),则往二叉树的左子树中查找。
如果当前节点值小于目标值(要查找的值),则往二叉树的右子树中查找。
//按值查找节点(递归)
public TreeNode findNode(TreeNode rootTreeNode, int val) {
if (rootTreeNode == null) {
return null;
}
if (rootTreeNode.val == val) {
return rootTreeNode;
}
if (rootTreeNode.val > val) {
return findNode(rootTreeNode.left, val);
}
if (rootTreeNode.val < val) {
return findNode(rootTreeNode.right, val);
}
//表示没有找到值
return null;
}
//按值查找节点(非递归)
public TreeNode findNode2(TreeNode rootTreeNode, int val) {
TreeNode currentNode = rootTreeNode;
while (currentNode.val != val) {
if (currentNode.val>val) {
currentNode = currentNode.left;
}else {
currentNode = currentNode.right;
}
if (currentNode == null) {
//到了叶子节点还没找到则返回空
return null;
}
}
return currentNode;
}
删除节点
删除节点是二叉树操作中最复杂的。在删除之前首先要查找要删的节点,找到节点后,这个要删除的节点可能会有三种情况需要考虑。
1、该节点是叶子接待你,没有子节点
要删除叶子节点,只需要改变该节点的父节点的引用值,在指向该节点的引用设置为null就可以。
2、该接待你有一个子节点
改变父节点的引用,将其直接指向要删除节点的子节点。
3、该节点有两个子节点
要删除有两个子节点的接待你,就需要使用它的中序后继来代替该节点。
中序后继节点的找法:中续后继节点就是中序遍历结果中当前节点的后继节点。从当前节点的右子树开始,一直往右子树的左子节点找,一直到右子树的左节点的叶子结果,就为中序后继节点。
//按值删除节点
public boolean deleteNode(TreeNode rooTreeNode, int val) {
TreeNode currentNode = rooTreeNode;
TreeNode parentNode = rooTreeNode;
boolean isLeftChild = true;
//找到节点位置
while (currentNode.val != val) {
parentNode = currentNode;
if (currentNode.val>val) {
currentNode = currentNode.left;
isLeftChild = true;
}else {
currentNode = currentNode.right;
isLeftChild = false;
}
if (currentNode == null) {
return false;
}
}
//删除节点过程
//如果该节点没有子节点
if (currentNode.left==null && currentNode.right==null) {
if (currentNode == rooTreeNode) {
rooTreeNode = null;
}
//如果它是父节点的左子节点
if (isLeftChild) {
parentNode.left = null;
}else {
parentNode.right = null;
}
//如果有一个左子节点
} else if (currentNode.right==null) {
if (currentNode == rooTreeNode) {
rooTreeNode = currentNode.left;
}
//如果它又是父节点的左子节点
if (isLeftChild) {
parentNode.left = currentNode.left;
}else {//如果它是父节点右子节点
parentNode.right = currentNode.left;
}
//如果有一个右子节点
}else if (currentNode.left==null) {
if (currentNode == rooTreeNode) {
rooTreeNode = currentNode.right;
}
if (isLeftChild) {
parentNode.left = currentNode.right;
}else {
parentNode.right = currentNode.right;
}
//如果它左右子节点都存在
}else {
TreeNode successorNode = getSuccessor(currentNode); //找到该节点的中序后继节点
if (currentNode == rooTreeNode) {
rooTreeNode = successorNode;
}else if (isLeftChild) {
parentNode.left = successorNode;
}else {
parentNode.right = successorNode;
}
successorNode.left = currentNode.left;
}
return true;
}
//找到当前节点的中序后继节点
public TreeNode getSuccessor(TreeNode node) {
TreeNode successorNode =node; //找到的中序后继目标节点
TreeNode parentNode = node;//找到的中序后继节点的父节点
//从当前节点右子树开始
TreeNode currentNode = node.right;
while (currentNode!=null) {
parentNode = successorNode;
successorNode = currentNode;
currentNode = currentNode.left; //往右子树的左节点走
}
if (successorNode != node.right) {
parentNode.left = successorNode.right;
successorNode.right = node.right;
}
return successorNode;
}
二叉树的四种遍历方式后续解析...