【劍指Offer系列54】二叉搜索樹的第k大節點

題目

給定一棵二叉搜索樹,請找出其中第k大的節點。

示例 1:
輸入: root = [3,1,4,null,2], k = 1
3
/
1 4

2
輸出: 4

示例 2:
輸入: root = [5,3,6,2,4,null,null,1], k = 3
5
/
3 6
/
2 4
/
1
輸出: 4

限制:
1 ≤ k ≤ 二叉搜索樹元素個數

代碼

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
        # 採用DFS中的中序遍歷的倒序,即遞減序列,“右,根,左”
        self.res, self.k = None, k # 返回值,k
        
        def dfs(root):
            if not root: return # 父節點爲空,終止
            dfs(root.right) # 遞歸遍歷右兒子
            self.k-=1
            if self.k==0: self.res=root.val # 返回二叉搜索樹的第k大節點,終止遍歷
            if self.k>0: dfs(root.left) # 遞歸遍歷左兒子
        
        dfs(root)
        return self.res

C++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int res;
    
    int kthLargest(TreeNode* root, int k) {
        dfs(root,k);
        return res;
    }
    
    void dfs(TreeNode* root, int &k) {
        if (!root) return;
        dfs(root->right,k);
        k--;
        if (k==0) res=root->val;
        if (k>0) dfs(root->left,k);
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章