Leetcode 230 - Kth Smallest Element in a BST（dfs）

题意

求bst的第k小。

思路

利用BST的中序遍历是排好序的这个性质做一下就好了。

算法1

开了额外的$O(n)$空间。利用vector保存了一下中序遍历的结果。

算法2

我们可以这样考虑：在我们中序遍历的时候，每次从左儿子节点返回的时候，代表元素变大了一个。

即每次从左儿子返回时，k–。当k为0的时候，即为我们的Kth。可以结合下面的例子理解一下：

比如我们要求第5小，我们遍历的顺序为：(10, 5) -> (8 ,5) -> (7, 5) -> (1, 5)。

然后到了1开始返回，我们每次从左节点返回一次，k减小一个。

于是，接下来为(1, 4) -> (7, 3) -> (8, 2)。发现8的右键点有值，于是下一步：(9, 2)。

返回：(9, 1) -> (10, 0)。于是，我们的第5小为10。

细节

注意算法2里面k应该设为全局变量或者传引用

代码

//algorithm 1
/**
 * 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:
    vector<int> v;

    void inorder(TreeNode *p) {
        if (p == NULL) return;
        inorder(p->left);
        v.push_back(p->val);
        inorder(p->right);
    }

    int kthSmallest(TreeNode* root, int k) {
        inorder(root);  
        return v[k - 1];
    }
};

//algorithm 2
/**
 * 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 ans;
    void dfs(TreeNode *p, int &k) {
        if (!p) return;
        dfs(p->left, k);
        k--;
        if (!k) ans = p->val;
        dfs(p->right, k);
    }

    int kthSmallest(TreeNode* root, int k) {
        dfs(root, k);
        return ans;
    }
};