二叉搜索树的第k大节点

题目链接

思路

• 二叉搜索树,它或者是一棵空树,或者是具有下列性质的二叉树:
  • 若它的左子树不空,则左子树上所有结点的值均小于它的根节点的值;
  • 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
  • 它的左,右子树也分别为二叉搜索树
• 定义current和res;使用递归递归顺序为右->根->左
  • 如果递归至叶子节点结束递归
  • 每次递归current++;当current==k时;将当前节点的值赋值给res
  • 当current==k时结束递归
• 递归结束将res返回

java

    int current, res;

    public int kthLargest(TreeNode root, int k) {
        dfs(root, k);
        return res;
    }

    private void dfs(TreeNode root, int k) {
        if (root == null) {
            return;
        }
        dfs(root.right, k);
        if (k == current) {
            return;
        }
        current++;
        if (k == current) {
            res = root.val;
        }
        dfs(root.left, k);
    }

go

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

var current, res int

func kthLargest(root *TreeNode, k int) int {
	current, res = 0,0
	dfs(root, k)
	return res
}

func dfs(root *TreeNode, k int) {
	if root == nil {
		return
	}
	dfs(root.Right, k)
	if k == current {
		return
	}
	current++
	if k == current {
		res = root.Val
	}
	dfs(root.Left, k)
}