剑指offer——二叉搜索树的第k个结点

题目描述
给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8） 中，按结点数值大小顺序第三小结点的值为4。

思路：
二叉搜索树按照中序遍历的顺序打印出来正好就是排序好的顺序。
方法一：
设置一个数组，然后中序遍历二叉树，将节点依次加入到数组中

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
    vector<TreeNode*> sortNodes;
    void treeToVector(TreeNode* root) {
        if (!root)
            return ;
        treeToVector(root->left);
        sortNodes.push_back(root);
        treeToVector(root->right);
    }
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if (k == 0)
            return NULL;
        
        treeToVector(pRoot);
        if (sortNodes.size() >= k)
            return sortNodes[k-1];
        else
            return NULL;
    }

    
};

方法二：
中序遍历，并且设置一个全局变量index用来判断目前进行到第几个节点。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
    int index = 1;
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if (!pRoot || k == 0)
            return NULL;
        
        TreeNode* node;
        node = KthNode(pRoot->left, k);
        if (node)
            return node;
        if (index == k)
            return pRoot;
        else
            index++;
        node = KthNode(pRoot->right, k);
        if (node)
            return node;
        
        return NULL;
    }
};