LeetCode-129. 求根节点到叶节点数字之和-Java-medium

题目链接

法一（dfs）

    private int sum = 0;

    /**
     * 在dfs的过程中统计sum
     *
     * @param root
     * @param path
     */
    private void dfs(TreeNode root, int path) {
        if (root == null) {
            return;
        }
        path = (path * 10) + root.val; // 更新path
        if (root.left == null && root.right == null) { // 遇到叶结点
            sum += path; // sum增加从根节点到叶节点的路径所表示的数字
        }
        dfs(root.left, path);
        dfs(root.right, path);
    }

    /**
     * 法一（dfs）
     *
     * @param root
     * @return
     */
    public int sumNumbers(TreeNode root) {
        dfs(root, 0);
        return sum;
    }

本地测试

        /**
         * 129. 求根节点到叶节点数字之和
         */
        lay.showTitle(129);
        Solution129 sol129 = new Solution129();
        List<Integer> arr129 = Arrays.asList(4, 9, 0, 5, 1);
        TreeNode root129 = treeOpt.createTreeByLayerOrder(arr129);
        treeOpt.layerOrder(root129);
        System.out.println(sol129.sumNumbers(root129));