转换二叉树,以便每个节点在C ++中将所有节点的总和存储在其右子树中
在本教程中,我们将讨论一个转换二叉树的程序,以便每个节点将所有节点的总和存储在其右子树中。
为此,我们将提供一个二叉树。我们的任务是返回另一棵树,其中每个节点必须等于该节点及其右子树的总和。
示例
#include <bits/stdc++.h>
using namespace std;
//树的节点结构
struct Node {
int data;
Node *left, *right;
};
//创建一个新节点
struct Node* createNode(int item){
Node* temp = new Node;
temp->data = item;
temp->left = NULL;
temp->right = NULL;
return temp;
}
//创建新的二叉树
int rightsum_tree(Node* root){
if (!root)
return 0;
if (root->left == NULL && root->right == NULL)
return root->data;
//更改左/右子树的值
int rightsum = rightsum_tree(root->right);
int leftsum = rightsum_tree(root->left);
//添加右子树的总和
root->data += rightsum;
return root->data + leftsum;
}
//以有序模式遍历树
void inorder(struct Node* node){
if (node == NULL)
return;
inorder(node->left);
cout << node->data << " ";
inorder(node->right);
}
int main(){
struct Node* root = NULL;
root = createNode(1);
root->left = createNode(2);
root->right = createNode(3);
root->left->left = createNode(4);
root->left->right = createNode(5);
root->right->right = createNode(6);
rightsum_tree(root);
cout << "Updated Binary Tree :\n";
inorder(root);
return 0;
}输出结果
Updated Binary Tree : 4 7 5 10 9 6