C++中具有偶数值祖父节点的节点之和
假设我们有一棵二叉树,我们需要找到具有偶数值祖父节点的节点的值之和。(祖父节点是指节点的父节点的父节点,如果存在)。如果没有这样的节点具有偶数值祖父节点,则返回0。例如,如果树如下所示:
输出将是18。红色节点是具有偶数值祖父节点的节点,而蓝色节点是偶数值祖父节点。
为了解决这个问题,我们将遵循以下步骤:
- 定义一个名为parent的映射
- 定义一个名为solve()的方法,它将接收节点和父节点作为参数
- 如果节点为空,则返回
- 如果父节点不为空,并且父节点存在于parent映射中,且parent[par]不为0,并且parent[par]的值为偶数,则
- res := res + 节点的值
- parent[node] := par
- solve(节点的左子节点, 节点)
- solve(节点的右子节点, 节点)
- 在主方法中,设置res := 0,调用solve(root, Null),然后返回res
示例(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 res; map <TreeNode*, TreeNode*> parent; void solve(TreeNode* node, TreeNode* par = NULL){ if(!node)return; if(par && parent.count(par) && parent[par] && parent[par]->val % 2 == 0){ res += node->val; } parent[node] = par; solve(node->left, node); solve(node->right, node); } int sumEvenGrandparent(TreeNode* root) { res = 0; parent.clear(); solve(root); return res; } }; main(){ vector<int> v = {6,7,8,2,7,1,3,9,NULL,1,4,NULL,NULL,NULL,5}; TreeNode *root = make_tree(v); Solution ob; cout << (ob.sumEvenGrandparent(root)); }
输入
[6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出
18
广告