剑指 Offer 54. 二叉搜索树的第k大节点
思路:二叉搜索树的中序倒序是从大到小,所以只需要在中序倒序遍历的时候计数就行了。
只是没有考虑到代码的鲁棒性。(k值问题)
剑指 Offer 54. 二叉搜索树的第k大节点
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { int res; int k; public int kthLargest(TreeNode root, int k) { this.k = k; dfs(root); return res; } void dfs(TreeNode root){ //因为可能一个树有5个结点,找第6大的节点。 if(root == null || k <= 0) return; //中序遍历倒序,先右子节点->根节点->左子节点 dfs(root.right); //寻找第k大,先--在判断。 if(--k == 0){ res = root.val; return; } dfs(root.left); //最后不用return,因为开始判断了为空返回。 //return; } }