LeetCode-129. 求根到叶子节点数字之和-深度优先搜索

/** 129. 求根到叶子节点数字之和

* @author 作者 Your-Name:

* @version 创建时间:2020年3月3日 下午4:34:48

* 给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。

例如,从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

示例 1:

输入: [1,2,3]
    1
   / \
  2   3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此,数字总和 = 12 + 13 = 25.

示例 2:

输入: [4,9,0,5,1]
    4
   / \
  9   0
 / \
5   1
输出: 1026
解释:
从根到叶子节点路径 4->9->5 代表数字 495.
从根到叶子节点路径 4->9->1 代表数字 491.
从根到叶子节点路径 4->0 代表数字 40.
因此,数字总和 = 495 + 491 + 40 = 1026.


*/

public class 求根到叶子节点数字之和 {
	public class TreeNode {
	      int val;
	      TreeNode left;
	      TreeNode right;
	      TreeNode(int x) { val = x; }
	  }
	List<String> list = new ArrayList<>();
	int sum=0;
	String str="";
	public int sumNumbers(TreeNode root) {
		dfs(root,str);
		for(String s:list)
		{
			sum+=Integer.valueOf(s);
		}
		return sum;
    }
	public List<String> dfs(TreeNode root,String str)
	{
		if(root==null)
			return list;
		str+=root.val;
		if(root.left==null&&root.right==null)
			list.add(str);
		dfs(root.left,str);
		dfs(root.right,str);
		return list;
	}
}
发布了72 篇原创文章 · 获赞 7 · 访问量 4095

猜你喜欢

转载自blog.csdn.net/l769440473/article/details/104669953