算法学习之二叉搜索树的第k个结点

题目描述

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

思路：直接用中序遍历俩获取递增的有序列表就可以了

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.*;
public class Solution {
    ArrayList<TreeNode> list = new ArrayList<>();
    TreeNode KthNode(TreeNode pRoot, int k)
    {
      addNode(pRoot);  
      if(k>=1&&list.size()>=k){
          return list.get(k-1);
      }
      return null;  
    }
    
    void addNode(TreeNode cur){
        if(cur!=null){
           addNode(cur.left); 
           list.add(cur); 
           addNode(cur.right); 
        }
    }


}

• 时间复杂度：O(n)。
• 空间复杂度：O(logn)。