2019-03-26 待提高

1.#### 二叉搜索樹中第K小的元素
給定一個(gè)二叉搜索樹,編寫一個(gè)函數(shù) kthSmallest 來查找其中第 k 個(gè)最小的元素。

說明:
你可以假設(shè) k 總是有效的,1 ≤ k ≤ 二叉搜索樹元素個(gè)數(shù)。

示例 1:

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

2
輸出: 1
示例 2:

輸入: root = [5,3,6,2,4,null,null,1], k = 3
5
/
3 6
/
2 4
/
1
輸出: 3
進(jìn)階:
如果二叉搜索樹經(jīng)常被修改(插入/刪除操作)并且你需要頻繁地查找第 k 小的值,你將如何優(yōu)化 kthSmallest 函數(shù)?

/**
 * 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 kthSmallest(TreeNode* root, int k) {
        
        int num = 0;
        int res ;
        findk( root , k , num , res);
        return res;
    }
    
    void findk(TreeNode* root, int k, int& num , int& res)
    {
        if(root == NULL) return;
        
        findk(root->left,k,num,res);
        
        num++;
        if(num == k ) res = root->val;
        
        findk(root->right,k,num,res);
        
        return;
        
    }
};
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容