《程序员面试金典》输出单层节点
生活随笔
收集整理的這篇文章主要介紹了
《程序员面试金典》输出单层节点
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目:對于一棵二叉樹,請設計一個算法,創建含有某一深度上所有結點的鏈表。
給定二叉樹的根結點指針TreeNode* root,以及鏈表上結點的深度,請返回一個鏈表ListNode,代表該深度上所有結點的值,請按樹上從左往右的順序鏈接,保證深度不超過樹的高度,樹上結點的值為非負整數且不超過100000。
解析:該題目考察的知識點較多,有層次遍歷法求樹的深度,并且還考察了單鏈表的創建。但是我們逐個擊破的話,該題目只是把多個知識點融合在一起,并不難。解題思想就是用隊列來層次遍歷二叉樹,然后用集合list記錄每一層遍歷出來的節點值,當達到某一深度dep,就利用該層的節點值創建為單鏈表,并返回。
import java.util.*; /* public class ListNode {int val;ListNode next = null;ListNode(int val) {this.val = val;} }*/ /* public class TreeNode {int val = 0;TreeNode left = null;TreeNode right = null;public TreeNode(int val) {this.val = val;} }*/ public class TreeLevel {public ListNode getTreeLevel(TreeNode root, int dep) {ListNode head =null;//單鏈表表頭Queue<TreeNode> queue = new LinkedList<>();//隊列,層次遍歷二叉樹的節點的時候保存節點信息List<Integer> list = new ArrayList<>();//存儲一層的節點信息,if(root==null){return head;}queue.add(root);int current =1;//當前層的節點數int next=0;//下一層的節點數int level=0;//記錄層次遍歷的深度while (!queue.isEmpty()){TreeNode node = queue.poll();list.add(node.val);current--;if(node.left!=null){queue.add(node.left);next++;}if(node.right!=null){queue.add(node.right);next++;}if(current==0){level++;if(level==dep){//已經遍歷到所需要的深度了break;}current=next;next=0;list = new ArrayList<>();//沒有到需要的深度,需要繼續重新創建list收集下一層的節點信息}}head = new ListNode(list.get(0));//注意一般的題目head都帶值的for(int i=list.size()-1;i>=1;i--){//頭插法創建單鏈表返回ListNode node = new ListNode(list.get(i));node.next=head.next;head.next=node;}return head;} }總結
以上是生活随笔為你收集整理的《程序员面试金典》输出单层节点的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 《数据库SQL实战》获取所有部门当前ma
- 下一篇: 《leetcode》longest-co