//二叉树算法的设计的总路线:明确一个节点要做的事情,然后剩下的事抛给框架。
void traverse(TreeNode root) {
//root需要做的事情
traverse(root.left);
traverse(root.right);
}
//如何把所有二叉树的节点值加一
void traverse(TreeNode root) {
if (root == null) return;
root.val += 1;
traverse(root.left);
traverse(root.right);
}
//如何判断两棵二叉树是否完全相同
boolean isIdentical(TreeNode root1, TreeNode root2) {
if (root1 == null && root2 == null) return true;
if (root1 == null || root2 == null) return false;
if (root1.val != root2.val) return false;
return traverse(root1.left, root2.left);
return traverse(root1.right, root2.right);
}
//判断二叉搜索树的合法性
boolean isBST(TreeNode root) {
return isBSThelper(root.left) && isBSThelper(root.right);
}
boolean isBSThelper(TreeNode root, TreeNode min, TreeNode max) {
if (root == null) return true;
if (min != null && root.val <= min) return false;
if (max != null && root.val >= max) return false;
return isBSThelper(root.left, min, root) && isBSThelper(root.right, root, max);
}
//结合二分法实现在bst找一个数字是否存在
boolean isInBST(TreeNode root, int target) {
if (root == null) return false;
if (root.val == target)
return true;
if (root.val < target)
return isInBST(root.right, target);
if (root.val > target)
return isInBST(root.left, target);
// root 该做的事做完了,顺带把框架也完成了,妙
}
//在bst中插入一个数字
TreeNode Insert(TreeNode root, int target) {
if (root == null) return new TreeNode(val);
if (root.val < target)
root.right = Insert(root.right, target);
if (root.val > target)
root.left = Insert(root.left, target);
return root;
}
//在bst中删除一个数
TreeNode deleteNode(TreeNode root, int key) {
if (root == null) return null;
if (root.val == key) {
// 这两个 if 把情况 1 和 2 都正确处理了
if (root.left == null) return root.right;
if (root.right == null) return root.left;
// 处理情况 3
TreeNode minNode = getMin(root.right);
root.val = minNode.val;
root.right = deleteNode(root.right, minNode.val);
} else if (root.val > key) {
root.left = deleteNode(root.left, key);
} else if (root.val < key) {
root.right = deleteNode(root.right, key);
}
return root;
}
TreeNode getMin(TreeNode node) {
// BST 最左边的就是最小的
while (node.left != null) node = node.left;
return node;
}
1 二叉树算法设计的总路线:把当前节点要做的事做好,其他的交给递归框架,不用 当前节点操心。
2 如果当前节点会对下面的子节点有整体影响,可以通过辅助函数增长参数列表,借助参数传递信息。
3 在二叉树框架之上,扩展出一套 BST 遍历框架:
```java
void BST(TreeNode root, int target) {
if (root.val == target)
// 找到目标,做点什么
if (root.val < target)
BST(root.right, target);
if (root.val > target)
BST(root.left, target);
}