Sum Root to Leaf Numbers

最新推荐文章于 2020-08-21 10:38:58 发布

原创最新推荐文章于 2020-08-21 10:38:58 发布 · 913 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode #二叉树

LeetCode 专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种算法，用于计算二叉树中所有从根节点到叶子节点的路径所代表数字的总和。通过递归遍历二叉树，并使用vector记录节点值及其层级信息，最终计算出所有有效路径的数值总和。

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.

由于加到和里面的值跟其所处节点的层数有关系，所以需要记录这一信息，这里用了一个vector来存，因为顺序代表着以一种层次关系。然后用一个全局的sum来累加。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
   int sum = 0;
	void sums(TreeNode*root,vector<int>vec)
	{
		if (!root)
			return;
		vec.push_back(root->val);
		if (!root->left && !root->right)
		{
			int count = vec.size()-1;
			for (auto i : vec)
			{
				sum += pow(10.0,count)*i;
				--count;
			}
			return;
		}
		sums(root->left,vec);
		sums(root->right,vec);
	}

	int sumNumbers(TreeNode* root)
	{
		vector<int>vec;
		sums(root,vec);
		return sum;
	}
};