剑指Offer_62_二叉搜索树的第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;    }}/*实现*/public class Solution {    TreeNode KthNode(TreeNode pRoot, int k)    {        Result re = kThNode(pRoot, k, 0);        return re.node;    }    private Result kThNode(TreeNode pRoot, int k, int cur) {        if (pRoot == null) return new Result(null, cur);        Result rl = kThNode(pRoot.left, k , cur);        if (rl.k == k) return rl;        cur = rl.k + 1;        if (cur == k) return new Result(pRoot, k);        Result rr =  kThNode(pRoot.right, k, cur);        if (rr.k == k) return rr;        return new Result(null,rr.k);    }    private class Result{        TreeNode node;        int k; //第几个数据        public Result(TreeNode node, int k){            this.node = node;            this.k = k;        }    }}