java算法训练------ 腾讯精选练习 50 题------ 二叉树的最近公共祖先、二叉搜索树中第K小的元素

二叉树的最近公共祖先
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x
的深度尽可能大（一个节点也可以是它自己的祖先）。

示例 1：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出：3
解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出：5
解释：节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。

示例 3：

输入：root = [1,2], p = 1, q = 2
输出：1

提示：

树中节点数目在范围 [2, 105] 内。
-109 <= Node.val <= 109
所有 Node.val 互不相同 。
p != q
p 和 q 均存在于给定的二叉树中。

思路：对于树这种数据结构第一就是想到递归，那怎么办去使用就是关键了。我们可以在题目中了解到，他需要的是找到的是两个子节点的最近的那一个父节点。所以我们就需要将节点遍历到两个子节点，并且找出两个子节点各自的第一个父节点，依次循环上去，如果没有找到子节点，那么就退出此子树。继续遍历其他的子树，最终只有三种情况：

1. 两个子节点都在最大的左子树
2. 两个子节点都在最大的右子树
3. 两个子节点分别在最大的左右子树
   那么我们就可一写出跳出递归的条件

if (root == null || root == p || root == q)
     return root;
if (left == null)
    return right;
if (right == null)
    return left;

接下来就只需要使用变量记录左右子树即可
最终代码：

public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    if (root == null || root == p || root == q)
        return root;
    TreeNode left = lowestCommonAncestor(root.left, p, q);
    TreeNode right = lowestCommonAncestor(root.right, p, q);
    if (left == null)
        return right;
    if (right == null)
        return left;
    return root;
}

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree

---

---

二叉搜索树中第K小的元素
给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 个最小元素（从 1 开始计数）。

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

示例 2：

输入：root = [5,3,6,2,4,null,null,1], k = 3
输出：3

提示：

树中的节点数为 n 。
1 <= k <= n <= 104
0 <= Node.val <= 104

进阶：如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化算法？

思路：把树遍历一遍，然后在对数据排序。
但是这种方法，会在排序非常耗时，因为他是二叉树，所以符合左节点<根节点<右节点，所以我们在遍历的时候，直接把先将左节点遍历到底，然后再去右节点，这样就可以保证我们遍历添加的数据就是从小到大排序好的数据。
代码：

public int kthSmallest(TreeNode root, int k) {
    ArrayList<Integer> list = new ArrayList<>();
    addNode(root, list);
    return list.get(k - 1);
}

public ArrayList<Integer> addNode(TreeNode root, ArrayList<Integer> list) {
    if (root == null) {
        return list;
    }
    addNode(root.left, list);
    list.add(root.val);
    addNode(root.right, list);
    return list;
}

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst