码迷,mamicode.com
首页 > 其他好文 > 详细

366. Find Leaves of Binary Tree

时间:2016-09-23 06:39:13      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:

Given a binary tree, collect a tree‘s nodes as if you were doing this: Collect and remove all leaves, repeat until the tree is empty.

Example:
Given binary tree 

          1
         /         2   3
       / \     
      4   5    

 

Returns [4, 5, 3], [2], [1].

Explanation:

1. Removing the leaves [4, 5, 3] would result in this tree:

          1
         / 
        2          

 

2. Now removing the leaf [2] would result in this tree:

          1          

 

3. Now removing the leaf [1] would result in the empty tree:

          []         

 

 

Returns [4, 5, 3], [2], [1].

 

 

public IList<IList<int>> FindLeaves(TreeNode root) {
        var res = new List<IList<int>>();
        
        if(root == null) return res;
       DFS(root,res);
       
        return res;
    }
    
    public int DFS(TreeNode root,IList<IList<int>> res )
    {
        if(root == null) return -1;
        int a = 1+Math.Max(DFS(root.left,res), DFS(root.right,res));
        if(a >= res.Count()) res.Add(new List<int>());
        res[a].Add(root.val);
        return a;
    }

 

 

public IList<IList<int>> FindLeaves(TreeNode root) {
        var res = new List<IList<int>>();
        if(root == null) return res;
        while(root != null)
        {
            var cur = new List<int>(); 
            root = DFS(root,cur);
            res.Add(cur);
        }
        return res;
    }
    
    public TreeNode DFS(TreeNode root,IList<int> res )
    {
        if(root == null) return null;
        if(root.left == null && root.right == null)
        {
            res.Add(root.val);
            return null;
        }
        root.left = DFS(root.left, res);
        root.right = DFS(root.right,res);
        return root;
    }

 

366. Find Leaves of Binary Tree

标签:

原文地址:http://www.cnblogs.com/renyualbert/p/5898560.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!