剑指offer-面试题63-二叉搜索树的第k个结点

二叉树节点定义：

package case63_KthNodeOfBST;

/**
* 二叉搜索树的结点结构定义
*
* @author WangSai
*
*/
public class MyNode {
int data;
MyNode lchild;
MyNode rchild;

public MyNode() {
}

public MyNode(int data) {
this.data = data;
}

}


代码实现：

package case63_KthNodeOfBST;

/**
* 题目：给定一棵二叉搜索树，请找出其中的第K大的结点。例如在下图的二叉搜索树中，按节点数值大小顺序第三个结点的值是4
* ............5...............................................................
* ........../....\............................................................
* .........3......7...........................................................
* ......./..\.../...\..........................................................
* ......2...4...6....8.........................................................
*
*
* @author WangSai
*
*/

public class KthNodeOfBST {

public static void main(String[] args) {
KthNodeOfBST kno = new Kth
4000
NodeOfBST();
MyNode root = new MyNode(5);
MyNode N2 = new MyNode(3);
MyNode N3 = new MyNode(7);
MyNode N4 = new MyNode(2);
MyNode N5 = new MyNode(4);
MyNode N6 = new MyNode(6);
MyNode N7 = new MyNode(8);
root.lchild = N2;
root.rchild = N3;
N2.lchild = N4;
N2.rchild = N5;
N3.lchild = N6;
N3.rchild = N7;
System.out.println(kno.findKthOfBST(root, 6).data);

}

/**
* 二叉搜索树的左子节<父节点<右子节点，根据中序遍历，即可进行排序并获取第K个节点。
*
* @param root，二叉搜索树的根节点
* @param k，k值
* @return 第k个二叉树节点
*/
private MyNode findKthOfBST(MyNode root, int k) {
// 异常值检测
if (root == null || k <= 0)
throw new IllegalArgumentException("非法输入参数，请重新检查...");
int[] temp = { k };
// 采用中序遍历的方法，便可以得到第K大的值
return findKthOfBSTCore(root, temp);
}

// 通过中序遍历二叉树
private MyNode findKthOfBSTCore(MyNode root, int[] temp) {
MyNode tempNode = null;
if (root == null)
return null;
// 先在左子树中遍历
tempNode = findKthOfBSTCore(root.lchild, temp);
// 左子树中没有找到
if (tempNode == null) {
// 当前的根结点是所要找的结点
if (temp[0] == 1)
return root;
// 当前的根结点不是要找的结点，但是已经找过了，所以计数器减一
else
temp[0]--;
}
// 左子树中没有找到，并且当前节点不是所要找的点，寻找右子节点
if (tempNode == null)
tempNode = findKthOfBSTCore(root.rchild, temp);
return tempNode;
}

}