《剑指offer》给定一颗二叉搜索树,请找出其中的第k大的结点。
生活随笔
收集整理的這篇文章主要介紹了
《剑指offer》给定一颗二叉搜索树,请找出其中的第k大的结点。
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目:給定一顆二叉搜索樹,請找出其中的第k大的結點。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按結點數值大小順序第三個結點的值為4。
解析:看到我寫的下面的代碼,像一篇文章一樣的長,我也是醉了,先提交上去了。思想就是先來個層次遍歷二叉樹,得到所有的節點的值,然后再排序,取到第k大的節點的值。當然這個只是值,而不是節點,需要得到具體的節點信息,我們需要再重新遍歷下該二叉樹,得到節點值為該值的節點,然后返回。
/* public class TreeNode {int val = 0;TreeNode left = null;TreeNode right = null;public TreeNode(int val) {this.val = val;} } */ import java.util.*; public class Solution {TreeNode KthNode(TreeNode pRoot, int k){if(k==0){return null;}List<Integer> list = new ArrayList<>();//存節點的值Queue<TreeNode> queue = new LinkedList<>();//存節點if(pRoot==null){return null;}queue.add(pRoot);int current =1;int next=0;while (!queue.isEmpty()){//層次遍歷二叉樹,然后得到所有節點的值TreeNode node=queue.poll();current--;list.add(node.val);if(node.left!=null){queue.add(node.left);next++;}if(node.right!=null){queue.add(node.right);next++;}if(current==0){current=next;next=0;}}Collections.sort(list);//給遍歷的二叉樹值的list排序,得到第k大的節點信息if(k>list.size()){return null;}int num=list.get(k-1);queue.clear();queue.add(pRoot);current=1;next=0;TreeNode result=null;while (!queue.isEmpty()){TreeNode node=queue.poll();current--;if(node.val==num){//找到第k大的節點信息,然后返回result=node;return result;}if(node.left!=null){queue.add(node.left);next++;}if(node.right!=null){queue.add(node.right);next++;}if(current==0){current=next;next=0;}}return result;} }總結
以上是生活随笔為你收集整理的《剑指offer》给定一颗二叉搜索树,请找出其中的第k大的结点。的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 《美团点评》编程题---有一个二维数组(
- 下一篇: 《剑指offer》序列化二叉树