给定一棵二叉搜索树,请找出其中第k大的节点。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我们知道二叉搜索树的中序遍历的从小到大的顺序,如何改写算法,称为从大到小的顺序尼?显然我们可以按照先右后左的方法改写中序遍历来达到从大到小的顺序。在每次遍历操作的时候进行k--,然后统计当前为第k个最大值即可。
package JianZhiOffer54; /** * @author :dazhu * @date :Created in 2020/4/9 11:30 * @description:二叉搜索树的第k大节点 * @modified By: * @version: $ */ public class Main { public static void main(String[]args){ TreeNode n1 = new TreeNode(1); TreeNode n2 = new TreeNode(2); TreeNode n3 = new TreeNode(3); TreeNode n4 = new TreeNode(4); TreeNode n5 = new TreeNode(5); TreeNode n6 = new TreeNode(6); n5.right = n6; n5.left = n3; n3.right = n4; n3.left = n2; n2.left = n1; Solution solution = new Solution(); System.out.println(solution.kthLargest(n5,1)); } } class Solution { private int largestNumber; //已经找到标志位,false没有找到,true已经找到 private boolean flag = false; private int k ; public int kthLargest(TreeNode root, int k) { this.k = k; recur(root); return largestNumber; } public void recur(TreeNode node) { if(flag == true){ return; } if(node == null){ return; } //右边的中序遍历 recur(node.right); if(k == 1&&flag ==false){ //k == 1意味着已经遍历k-1次,当前位置就是目标位置 largestNumber = node.val; flag = true; return; } k--; recur(node.left); } } class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int x) { val = x; } }
原文:https://www.cnblogs.com/dazhu123/p/12666060.html