62.二叉树搜索树的第K个节点——剑指offer

weixin_30416871

于 2019-07-09 21:50:00 发布

阅读量53

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/developing/articles/11160710.html

博客给出了求解二叉树搜索树第K个节点的代码。定义了TreeNode指针类型，通过深度优先搜索（DFS）遍历二叉树，在中序遍历过程中找到第K个节点。代码实现了KthNode函数来完成此功能。

typedef TreeNode* pnode;
class Solution {
    int m;
    pnode ans;
    void dfs(pnode p){
        if(!p || m < 1) return;
        dfs(p -> left);
        if(m == 1) ans = p;
        --m;
        if(m > 0) dfs(p -> right);
    }
public:
    TreeNode* KthNode(TreeNode* p, unsigned int k){
        ans = NULL; m = k;
        dfs(p);
        return ans; 
    }
};