二叉搜索树概念:
(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
举例中的二叉树实际结构:
核心思想:中序遍历,也就是 左 根 右 ,二叉树变成了2,3,4,5,6,7,8这样的顺序。查找第k个元素即可。
第一种办法:递归中序遍历+利用数组存储二叉树数据。再获取第k个元素。
static ArrayList<Node> list =new ArrayList<>();
//前序遍历 根左右
//中序遍历 左根右
//后序遍历 左右根
static Node KthNode(Node pRoot, int k) {
midTraversal(pRoot);
if (k<=0||k>list.size())
return null;
return list.get(k-1);
}
public static void main(String[] args) {
Node node1=new Node(5);
Node node2=new Node(3);
Node node3=new Node(7);
node1.left=node2;
node1.right=node3;
midTraversal(node1);
for (Node node:list) {
System.out.println(node.val);
}
System.out.println(KthNode(node1, 1).val);
}
public static void midTraversal(Node node){
if (node==null)
return;
midTraversal(node.left);
list.add(node);
midTraversal(node.right);
}
第二种办法:栈实现
static Node KthNode(Node pRoot, int k) {
if (k<=0||pRoot==null)
return null; //合法性判断
int index=0;//index用来判断是否是第k个元素
Node tmp=pRoot;
Stack<Node> stack=new Stack<>(); //利用栈的特点,核心思想就是先跑到最左边的节点,从该节点开始遍历
while (!stack.empty()||tmp!=null){//循环跳出的条件必须是栈为空并且该节点已经是空节点了
if (tmp!=null){
stack.push(tmp);//始终把左节点放入栈
tmp=tmp.left;
}else {
tmp=stack.pop();//如果没有左节点,则开始比较,如果当前节点index等于k,说明已经找到了。
index++;
if (index==k){
return tmp;
}
tmp=tmp.right;//如果到这还没有返回,说明index还没有到达k个元素,需要比较当前节点的右孩子
}
}
return null; //不可能走到此语句
}