博客
关于我
【Lintcode】650. Find Leaves of Binary Tree
阅读量:182 次
发布时间:2019-02-28

本文共 1448 字,大约阅读时间需要 4 分钟。

题目地址:

给定一个二叉树,要求返回其叶子节点,然后将叶子节点删去,继续返回叶子节点,继续删去叶子,直到树空。返回每次得到的叶子节点。

更快的 O ( n ) O(n) O(n)的方法参考。下面介绍一个逐次删掉叶子的方法:

思路是DFS,开一个函数对二叉树做DFS,每次DFS的时候,将叶子收集在一个列表里,同时删除掉叶子节点;删除的方式是,将这个函数的返回值设为删掉所有叶子后的二叉树的树根,这样就可以在到达叶子的时候返回null回去连到上一层的树根上,就达到了删掉叶子的效果。代码如下:

import java.util.ArrayList;import java.util.List;public class Solution {       /*     * @param root: the root of binary tree     * @return: collect and remove all leaves     */    public List
> findLeaves(TreeNode root) { // write your code here List
> res = new ArrayList<>(); while (root != null) { List
cur = new ArrayList<>(); root = dfs(root, cur); res.add(cur); } return res; } private TreeNode dfs(TreeNode root, List
cur) { if (root == null) { return null; } else if (root.left == null && root.right == null) { cur.add(root.val); return null; } else { root.left = dfs(root.left, cur); root.right = dfs(root.right, cur); return root; } }}class TreeNode { int val; TreeNode left, right; TreeNode(int x) { val = x; }}

时间复杂度 O ( n 2 ) O(n^2) O(n2)(最差情况是每次删除的叶子节点个数是 1 1 1,比如整个二叉树是个链表,那么此时时间复杂度最高。可以考虑 T ( n ) = ∑ i = 1 h O ( n − l i ) T(n)=\sum_{i=1}^{h}O(n-l_i) T(n)=i=1hO(nli),其中 l i l_i li是每次删除的叶子, h h h是树的高度,容易知道最差情况是二叉树是链表的时候。而若二叉树较为平衡时,时间复杂度可以降到 O ( n log ⁡ n ) O(n\log n) O(nlogn)),空间 O ( h ) O(h) O(h)(除去最后返回的列表的空间)。

转载地址:http://idds.baihongyu.com/

你可能感兴趣的文章
Net 应用程序如何在32位操作系统下申请超过2G的内存
查看>>
Net.Framework概述
查看>>
NET3.0+中使软件发出声音[整理篇]<转>
查看>>
net::err_aborted 错误码 404
查看>>
NetApp凭借领先的混合云数据与服务把握数字化转型机遇
查看>>
NetAssist网络调试工具使用指南 (附NetAssist工具包)
查看>>
Netbeans 8.1启动参数配置
查看>>
NetBeans IDE8.0需要JDK1.7及以上版本
查看>>
NetBeans之JSP开发环境的搭建...
查看>>
NetBeans之改变难看的JSP脚本标签的背景色...
查看>>
netbeans生成的maven工程没有web.xml文件 如何新建
查看>>
netcat的端口转发功能的实现
查看>>