题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
题目分析
首先,我们可以先画图。画完图后我们要想办法从中找出第K小的节点。
因为这是二叉搜索树,我们可以轻易发现它的中序遍历序列就是从小到大排列,也就是我们可以直接中序遍历,同时计数,就可以得到我们想要的节点了。
不过需要注意的是我们的计数变量k应该在函数外面,不然递归进去后回来时是无法获得已经改变了的k值的。
代码
function KthNode(pRoot, k) { if(pRoot == null || k == 0){ return null; } //为了能追踪k,应该把KthNodeCore函数定义在这里面,k应该在KthNodeCore函数外面 function KthNodeCore(pRoot){ let target = null; if(pRoot.left != null){ target = KthNodeCore(pRoot.left, k) } if(target == null){ if(k == 1){ target = pRoot; } k--; } if(target == null && pRoot.right != null){ target = KthNodeCore(pRoot.right, k); } return target; } return KthNodeCore(pRoot); }
原文地址:https://www.cnblogs.com/wuguanglin/p/KthNode.html
时间: 2024-10-04 10:13:15