C++ 中最深叶子总和
假设我们有一棵二叉树,我们必须找出其最深叶节点之值的总和。因此,如果树形图如下:
则输出将为 15。
为了解决此问题,我们将按照以下步骤执行操作:
- 定义一个映射 m 和 maxDepth
- 定义一个递归方法 solve(),它会获取节点和级别,最初级别为 0
- 如果节点不存在,则返回
- maxDepth := 级别和 maxDepth 的最大值
- 将 m[level] 乘以节点值
- solve(节点左节点,级别 + 1)
- solve(节点右节点,级别 + 1)
- 在主方法中,设置 maxDepth := 0,然后解决 (root, 0)
- 返回 m[maxDepth]
示例 (C++)
让我们看看以下实现,以更好地理解它:
#include <bits/stdc++.h> using namespace std; class TreeNode{ public: int val; TreeNode *left, *right; TreeNode(int data){ val = data; left = NULL; right = NULL; } }; void insert(TreeNode **root, int val){ queue<TreeNode*> q; q.push(*root); while(q.size()){ TreeNode *temp = q.front(); q.pop(); if(!temp->left){ if(val != NULL) temp->left = new TreeNode(val); else temp->left = new TreeNode(0); return; } else{ q.push(temp->left); } if(!temp->right){ if(val != NULL) temp->right = new TreeNode(val); else temp->right = new TreeNode(0); return; } else{ q.push(temp->right); } } } TreeNode *make_tree(vector<int> v){ TreeNode *root = new TreeNode(v[0]); for(int i = 1; i<v.size(); i++){ insert(&root, v[i]); } return root; } class Solution { public: int maxDepth; map <int, int> m; void solve(TreeNode* node, int level = 0){ if(!node)return; maxDepth = max(level, maxDepth); m[level] += node->val; solve(node->left, level + 1); solve(node->right, level + 1); } int deepestLeavesSum(TreeNode* root) { maxDepth = 0; m.clear(); solve(root); //cout << maxDepth << endl; return m[maxDepth]; } }; main(){ vector<int> v = {1,2,3,4,5,NULL,6,7,NULL,NULL,NULL,NULL,8}; TreeNode *root = make_tree(v); Solution ob; cout << (ob.deepestLeavesSum(root)); }
输入
[1,2,3,4,5,null,6,7,null,null,null,null,8]
输出
15
广告