力扣hot100 二叉搜索树中第k小的元素分治中序遍历

最新推荐文章于 2024-08-13 14:47:10 发布

兑生

最新推荐文章于 2024-08-13 14:47:10 发布

阅读量479

点赞数 9

分类专栏： # 力扣 hot100 文章标签： leetcode 算法职场和发展

兑生：前世大厂架构师、上市公司CEO，被小人陷害，重生为25应届生，三连加关注，看我如何逆风翻盘。

本文链接：https://blog.csdn.net/lt6666678/article/details/135381892

版权

力扣 hot100 专栏收录该内容

100 篇文章 1 订阅

订阅专栏

👨‍🏫 题目地址

在这里插入图片描述
👨‍🏫 参考题解

😋 分治

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
 	public int kthSmallest(TreeNode root, int k)
	{
		int n = nodeCount(root.left);//计算当前树的左子树的结点数
		if (n + 1 == k)// 说明根节点就是第 k 小的数
			return root.val;
		else if (n + 1 < k)// 说明第 k 小的数在当前树的 右子树的第（k-n-1)小处
			return kthSmallest(root.right, k - n - 1);
		else// 说明第 k 小的数在当前树的 左子树的 第（k)小处
			return kthSmallest(root.left, k);

	}

	/**
	 * @param root 当前树的根节点
	 * @return 当前树的总结点数（包含根节点）
	 */
	int nodeCount(TreeNode root)
	{
		if (root == null)
			return 0;
		return 1 + nodeCount(root.left) + nodeCount(root.right);
	}
}

😋 递归中序遍历

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 * int val;
 * TreeNode left;
 * TreeNode right;
 * TreeNode() {}
 * TreeNode(int val) { this.val = val; }
 * TreeNode(int val, TreeNode left, TreeNode right) {
 * this.val = val;
 * this.left = left;
 * this.right = right;
 * }
 * }
 */
class Solution {
    int num = 0;
    int res;

    public int kthSmallest(TreeNode root, int k) {
        inorderTraversal(root, k);
        return res;
    }

    private void inorderTraversal(TreeNode node, int k) {
        if (node == null || num > k) {
            return;
        }
        inorderTraversal(node.left, k);
        num++;
        if (num == k) {
            res = node.val;
            return;
        }
        inorderTraversal(node.right, k);
    }
}