使用 C++ 转换二叉树,使每个节点都存储其右子树中所有节点的总和

c++server side programmingprogramming更新于 2024/10/3 0:22:00

在本教程中,我们将讨论一个转换二叉树的程序,使每个节点都存储其右子树中所有节点的总和。

为此,我们将提供一棵二叉树。我们的任务是返回另一棵树,其中每个节点必须等于节点与其右子树的总和。

示例

#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 rightssum_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 << "更新后的二叉树:\n";
   inorder(root);
   return 0;
}

输出

更新后的二叉树:
4 7 5 10 9 6

相关文章