刷題-力扣-230. 二叉搜尋樹中第K小的元素
阿新 • • 發佈:2021-06-27
230. 二叉搜尋樹中第K小的元素
題目連結
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
題目描述
給定一個二叉搜尋樹的根節點 root ,和一個整數 k ,請你設計一個演算法查詢其中第k個最小元素(從 1 開始計數)。
示例 1:
輸入:root = [3,1,4,null,2], k = 1
輸出:1
示例 2:
輸入:root = [5,3,6,2,4,null,null,1], k = 3
輸出:3
提示:
樹中的節點數為 n 。
1 <= k <= n <= 104
0 <= Node.val <= 104
進階:如果二叉搜尋樹經常被修改(插入/刪除操作)並且你需要頻繁地查詢第 k 小的值,你將如何優化演算法?
題目分析
- 根據題目分析返回二叉搜尋樹中第k大的節點值
- 使用中序遍歷,遍歷二叉搜尋樹,遍歷結果即按照遞增排序
程式碼
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { public: int kthSmallest(TreeNode* root, int k) { vector<int> nodeValVec; recursion(root, nodeValVec); return nodeValVec[k - 1]; } private: void recursion(TreeNode* root, vector<int>& nodeValVec) { if (!root) return; recursion(root->left, nodeValVec); nodeValVec.push_back(root->val); recursion(root->right, nodeValVec); } };