LeetCode#230.二叉搜索书中第k小的元素

二叉搜索树种第k小的元素：给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

三种解法：

1.第一种是要计算左子树的节点个数，然后来判断第k个节点在根还是左子树还是右子树
2.第二种是递归中序遍历，查找第k个结点
3.第三种是非递归中序遍历，查找第k个遍历到的节点

/**
 * 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) {
        
        思路：如果左子树是k-1个节点，那么第k个节点就是根
             如果左子树大于k个节点，那么第k个节点肯定在左子树，那就去左子树找
             如果左子树节点少于k个节点，那么第k个节点肯定在右子树,特别注意传递个右子树查找
             要减去左子树的长度和根的长度
        
        if(root == nullptr) return -1;
        int tmp = calculate(root->left);
        if(tmp == k-1){
            //如果左子树刚好是k-1，那么第k个节点一定是这个根节点
            return root->val;
        }else if(tmp >= k){
            //如果左子树大于等于k，去左子树查找，仍然查找第k个
            return kthSmallest(root->left,k);
        }else{
            //如果左子树少于k个节点，就得去右子树找第k-左子树个数的节点-1(根)
            return kthSmallest(root->right,k-tmp-1);
        }
    }
    int calculate(TreeNode* root){
        //该方法用来计算该子树有多少个节点
        if(root == nullptr) return 0;
        //返回左子树+右子树+根结点
        return calculate(root->left) + calculate(root->right) + 1;
    }*/
    /*int kthSmallest(TreeNode* root, int k) {
       
       或者使用一下中序遍历，然后查找到第k个就行，试一下吧
       
       int num = 0,ret;
       kthSmallestCore(root,k,num,ret);
       return ret;
    }
    void kthSmallestCore(TreeNode* root,int k,int& num,int& ret){
        //中序遍历
        if(root != nullptr){
            //第一次扫描到节点
            kthSmallestCore(root->left,k,num,ret);
            //中序的处理,第一个访问到的就是最小的节点
            num++;
            if(num == k){
                //如果这次是第k个访问的节点
                ret = root->val;
                return;
            }
            //第二次扫描到节点
            kthSmallestCore(root->right,k,num,ret);
        }
    }*/
    int kthSmallest(TreeNode* root, int k) {
       /*
       或者使用一下中序非递归遍历，然后查找到第k个就行，试一下吧
       */
       stack<TreeNode*> s;
       int count = 0;
       while(s.size()>0 || root!=nullptr){
           while(root){
               s.push(root);
               root = root->left;
           }
           if(s.size()>0){
               count++;
               if(count == k){
                   return s.top()->val;
               }
               root = s.top()->right;
               s.pop();     
           }
       }
       return -1;
    }

};```