题目地址：

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

更快的 $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)$ （最差情况是每次删除的叶子节点个数是 $1$ ，比如整个二叉树是个链表，那么此时时间复杂度最高。可以考虑 $T(n)=\sum_{i=1}^{h}O(n-l_i)$ ，其中 $l_i$ 是每次删除的叶子， $h$ 是树的高度，容易知道最差情况是二叉树是链表的时候。而若二叉树较为平衡时，时间复杂度可以降到 $O(n\log n)$ ），空间 $O (h)$ （除去最后返回的列表的空间）。

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

你可能感兴趣的文章