输出单层结点之程序员面试经典

最新推荐文章于 2020-10-24 23:32:35 发布

原创最新推荐文章于 2020-10-24 23:32:35 发布 · 329 阅读

0 ·

CC 4.0 BY-SA版权

程序员面试经典编程题专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种算法，用于将二叉树中特定深度的所有节点转化为链表形式。通过对二叉树进行层次遍历，并利用队列和ArrayList存储各层节点值，最后根据指定深度构建链表。

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

题目描述

对于一棵二叉树，请设计一个算法，创建含有某一深度上所有结点的链表。

给定二叉树的根结点指针TreeNode* root，以及链表上结点的深度，请返回一个链表ListNode，代表该深度上所有结点的值，请按树上从左往右的顺序链接，保证深度不超过树的高度，树上结点的值为非负整数且不超过100000。

思路：简单来说，步骤就是1.可以先用按层遍历把每一层都用ArrayList<Integer>去存储；2.通过深度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) {
       Queue<TreeNode> queue =new LinkedList<TreeNode>();
	        ArrayList<ArrayList<Integer>> lists =new ArrayList<ArrayList<Integer>>();
	        ArrayList<Integer> list =new ArrayList<Integer>();
	        queue.add(root);
	        int high=0;
	        TreeNode last=root;
	        TreeNode nlast=root;
	        while(!queue.isEmpty()){
	        	TreeNode temp =queue.poll();
	        	list.add(temp.val);
	        	if(temp.left!=null){
	        		queue.add(temp.left);
	        		nlast=temp.left;        		
	        	}
	        	if(temp.right!=null){
	        		queue.add(temp.right);
	        		nlast=temp.right;
	        	}
	        	if(temp==last){
	        		lists.add(list);
	        	    last=nlast;
	        	    list=new ArrayList<Integer>();
	        	    high++;
	        	}
	        }
	        
	        int[][] A=new int[high][];
	        for(int i=0;i<high;i++){
	        	A[i] =new int[lists.get(i).size()];
	        	for(int j=0;j<lists.get(i).size();j++){
	        		A[i][j]=lists.get(i).get(j);
	        	}
	        }
		  ListNode l1 =new ListNode(-1);
		  ListNode ln =l1;
		  ArrayList<Integer> aaa =lists.get(dep-1);
		  for (Integer num : aaa) {
			 ln.next=new ListNode(num);
			 ln=ln.next;
		}
		  return l1.next;
	 
    }
}