剑指offer:二叉搜索树的第k个节点

在这里插入图片描述
二叉搜索树概念:
(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
举例中的二叉树实际结构:
在这里插入图片描述
核心思想:中序遍历,也就是 左 根 右 ,二叉树变成了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; //不可能走到此语句

    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值