LeetCode129. 求根到叶子节点数字之和

阳光穿透心脏的1/2处 2022-05-11 09:30 173阅读 0赞

给定一个二叉树，它的每个结点都存放一个 `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.

--------------------

思路：深度优先遍历二叉树。

/**
     * Definition for a binary tree node.
     * public class TreeNode {
     *     int val;
     *     TreeNode left;
     *     TreeNode right;
     *     TreeNode(int x) { val = x; }
     * }
     */
    class Solution {
        public int sumNumbers(TreeNode root) {
            List<Integer> res=new LinkedList<Integer>();//保存根结点到叶子结点的所有路径值
            String tmp="";//临时存储根结点到叶子结点的路径值
            dfs(root,res,tmp);//深度优先遍历二叉树
            int sum=0;//所有路径和
            for(Integer i:res){
                sum+=i;
            }
            return sum;
        }
        public  void dfs(TreeNode root, List<Integer> res,String tmp){
            if(root==null){
                return ;
            }
            if(root.left==null&&root.right==null){  //叶子结点
                tmp+=root.val;
                int n=Integer.parseInt(tmp);
                res.add(n);
                return ;
            }
            tmp+=root.val;
            dfs(root.left,res,tmp);
            tmp.substring(0,tmp.length()-1);
            dfs(root.right,res,tmp);
            tmp.substring(0,tmp.length()-1);
        }
    }