【Leetcode】Sum Root to Leaf Numbers

给定一个只包含0到9的二叉树,每条根到叶的路径代表一个数字。例如,路径1->2代表数字12。求所有根到叶路径所代表的数字之和。对于示例输入,总和为12 + 13 = 25。可以采用两种方法解决:1) 保存所有路径并累加;2) 在递归过程中处理路径和。
摘要由CSDN通过智能技术生成

题目链接:https://leetcode.com/problems/sum-root-to-leaf-numbers/

题目:

Given a binary tree containing digits from 0-9 only, each root-to-leaf path could represent a number.

An example is the root-to-leaf path 1->2->3 which represents the number 123.

Find the total sum of all root-to-leaf numbers.

For example,

    1
   / \
  2   3

The root-to-leaf path 1->2 represents the number 12.
The root-to-leaf path 1->3 represents the number 13.

Return the sum = 12 + 13 = 25.

思路:

1、参考Binary Tree Paths 中保存所有从根到叶路径的方法,对所有路径处理一下再相加就能得到结果。这种方法是思路简单,利用 可以保存从根到叶所有路径的方法可以解决很多类似的root-leaf问题。不过需要O(n)的空间保存所有路径。

2、在递归过程中处理路径和。优点是算法简洁,不需要保存路径,空间复杂度O(1)。

算法1:

	List<String> strList = new ArrayList<String>();

	public int sumNumbers(TreeNode root) {
		int sum = 0;
		if (root != null) {
			findPath(root, String.valueOf(root.val));
			for (int i = 0; i < strList.size(); i++) {
				sum += getConbinaFromStr(strList.get(i));
			}
		}
		return sum;
	}

	public void findPath(TreeNode p, String path) {
		if (p.left == null && p.right == null) {
			strList.add(path);
		}
		if (p.left != null)
			findPath(p.left, path + "->" + p.left.val);
		if (p.right != null)
			findPath(p.right, path + "->" + p.right.val);
	}

	public int getConbinaFromStr(String str) {
		int sum = 0;
		String s[] = str.split("->");
		for (int i = 0; i < s.length; i++) {
			int tmp = Integer.parseInt(s[i]);
			sum = sum * 10 + tmp;
		}
		return sum;
	}


算法2:

	int sum = 0;

	public int sumNumbers2(TreeNode root) {
		addPSum(root, 0);
		return sum;
	}

	public void addPSum(TreeNode root, int curNum) {
		if (root == null)
			return;
		curNum = curNum * 10 + root.val; // 此时该路径的和
		if (root.left == null && root.right == null) { // 从根走到叶了
			sum += curNum;
			return;
		}
		if (root.left != null) {
			addPSum(root.left, curNum);
		}
		if (root.right != null) {
			addPSum(root.right, curNum);
		}
	}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值