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