二叉搜索树第K大节点和第K小节点

最新推荐文章于 2022-04-29 11:01:16 发布

qy_zhizi

最新推荐文章于 2022-04-29 11:01:16 发布

阅读量414

点赞数

分类专栏：基础算法文章标签：二叉树 dfs leetcode 二叉搜索树数据结构

本文链接：https://blog.csdn.net/qq_36275734/article/details/108564819

版权

基础算法专栏收录该内容

13 篇文章 0 订阅

订阅专栏

第K大节点

dfs,中序遍历先便利右子树,再遍历左子树,如果中序遍历是升序的情况.

在这里插入图片描述

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

        self.k = k
        dfs(root)
        return self.res

第k小节点(牛客)

在这里插入图片描述

dfs,中序遍历先便利左子树,再遍历右子树,如果中序遍历是升序的情况.

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    def KthNode(self, pRoot, k):
        # write code here
        self.k = k
        self.res = None
        def dfs(root):
            if not root:
                return
            dfs(root.left)
            if self.k == 0:
                return
            self.k -=1
            if self.k == 0: 
                self.res = root
                return
            dfs(root.right)
        dfs(pRoot)
        return self.res