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

题目地址：

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

更快的$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/