转换二叉树,以便每个节点在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