剑指笔记——54.二叉树的第K大节点

最新推荐文章于 2022-10-21 15:17:35 发布

chenxy132

最新推荐文章于 2022-10-21 15:17:35 发布

阅读量295

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指

本文链接：https://blog.csdn.net/chenxy132/article/details/89668937

剑指专栏收录该内容

52 篇文章

订阅专栏

这篇博客介绍了如何在二叉搜索树中找到第K大的节点。通过中序遍历的方式，利用栈辅助实现，当计数达到K时返回对应的节点。提供了两种不同的代码实现，一种基于官方思路，另一种为博主自己的实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：给定一棵二叉搜索树，请找出其中第K大的节点。

思路：按照剑指中的思路可知，对于二叉搜索树来说，中序遍历是递增排序的，我们只需要求出该二叉树的中序遍历，在找到第K个节点就可以了。

代码来源：https://blog.csdn.net/snow_7/article/details/51926574

代码：public TreeNode KthNode(TreeNode pRoot, int k)
        {
            if(pRoot == null || k <= 0) //如果根节点为空或者给定的K值小于0
                return null;
            TreeNode current = pRoot;
            TreeNode kNode = null;//用来保存第K个节点
            int count = 0;
            LinkedList<TreeNode> stack = new LinkedList<TreeNode>(); //存放节点
            //中序遍历二叉搜索树
            while(!stack.isEmpty() || current != null) //当栈不为空或者当前节点不为空
           {
              while(current!=null)
             {
               stack.push(current); //把当前节点加入栈，
               current = current.left; //查看左节点
             }
              if(!stack.isEmpty())
              {
                  current = stack.pop();//每出栈一个节点让count++;对比count和K的值，第k个节点时跳出循环返回节点，
                  count++;
                  if(count == k)
                  {
                      kNode = current;
                      break;
                  }
                  current = current.right; //查看右节点
              }
           }
            return kNode;
        }

下面这个是我自己写的代码：

import java.util.*;
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null||k<=0){
            return null;
        }
        Stack<TreeNode> s=new Stack<>();
        while(!s.isEmpty()||pRoot!=null){
            while(pRoot!=null){
                s.push(pRoot);
                pRoot=pRoot.left;
            }
            if(!s.isEmpty()){
                TreeNode temp=s.pop();
              pRoot=temp.right;
                k--;
                if(k==0){
                    return temp;
                }
            }
        }
        return null;
    }

}