博客
关于我
【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/

你可能感兴趣的文章
MySQL-【4】基本操作
查看>>
Mysql-丢失更新
查看>>
Mysql-事务阻塞
查看>>
Mysql-存储引擎
查看>>
mysql-开启慢查询&所有操作记录日志
查看>>
MySQL-数据目录
查看>>
MySQL-数据页的结构
查看>>
MySQL-架构篇
查看>>
MySQL-索引的分类(聚簇索引、二级索引、联合索引)
查看>>
Mysql-触发器及创建触发器失败原因
查看>>
MySQL-连接
查看>>
mysql-递归查询(二)
查看>>
MySQL5.1安装
查看>>
mysql5.5和5.6版本间的坑
查看>>
mysql5.5最简安装教程
查看>>
mysql5.6 TIME,DATETIME,TIMESTAMP
查看>>
mysql5.6.21重置数据库的root密码
查看>>
Mysql5.6主从复制-基于binlog
查看>>
MySQL5.6忘记root密码(win平台)
查看>>
MySQL5.6的Linux安装shell脚本之二进制安装(一)
查看>>