牛客NC81 二叉搜索树的第k个节点【中等中序遍历】

最新推荐文章于 2024-08-19 11:12:00 发布

赵长辉

最新推荐文章于 2024-08-19 11:12:00 发布

阅读量386

点赞数 9

文章标签：算法数据结构

本文链接：https://blog.csdn.net/weixin_37991016/article/details/135737959

版权

这篇文章介绍了如何使用Java编程语言，通过非递归的方式实现中序遍历二叉树，找到第k个节点的值。关键步骤包括使用栈来存储遍历过程中的节点，并记录遍历计数。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

在这里插入图片描述
题目链接：
https://www.nowcoder.com/practice/57aa0bab91884a10b5136ca2c087f8ff

思路

中序遍历

参考答案

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 proot, int k) {
        if (proot == null) return -1;

        //利用非递归中序遍历思路来做
        Stack<TreeNode> stk = new Stack<>();
        int cnt = 0;
        while (!stk.isEmpty() || proot != null) {
            while (proot != null) {
                stk.add(proot);
                proot = proot.left;
            }

            proot = stk.pop();
            cnt++;
            if (cnt == k) return proot.val;

            proot = proot.right;

        }

        return -1;
    }
}