牛客网：NC81 二叉搜索树的第k个节点

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param proot TreeNode类 
     * @param k int整型 
     * @return int整型
     */
    public int KthNode (TreeNode root, int k) {
        if (root == null || k == 0){
            return - 1;
        }
        // write code here
        List<Integer> list = new ArrayList();
        dfs(root, list);
        Collections.sort(list);
        return list.size() < k  ? -1 : list.get(k - 1);
    }
    
    private void dfs (TreeNode root, List<Integer> list ){
        if (root == null){
            return;
        }
        list.add(root.val);
        dfs(root.left, list);
        dfs(root.right, list);
    }
}

官方题解：使用递归，中序遍历

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param proot TreeNode类 
     * @param k int整型 
     * @return int整型
     */
    // 题解
    // 首先他是二叉搜索树，所以他是有序的
    // 设置全局变量记录他的值
    int count = 0, res = -1;
    public int KthNode (TreeNode root, int k) {
        // write code here
        dfs(root, k);
        return res;
    }
    
    private void dfs (TreeNode root, int k){
        if (root == null || count >= k){
            return;
        }
        dfs(root.left, k);
        count++;
        if (count == k){
            res = root.val;
        }
        dfs(root.right, k);
    }
}