leetcode刷题笔记31

面试题54. 二叉搜索树的第k大节点

难度简单13给定一棵二叉搜索树,请找出其中第k大的节点。

class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
    self.res,self.count = 0,k
    def dfs(root):
    	if not root: return None
    	else: 
    		dfs(root.right)
    		self.count -= 1
    		if self.count == 0 : self.res = root.val
    		else: dfs(root.left)
    	dfs(root)
    	return self.res

你可能感兴趣的:(leetcode)