从二叉搜索树到更大和树

问题描述

给出二叉 搜索 树的根节点，该二叉树的节点值各不相同，修改二叉树，使每个节点 node 的新值等于原树中大于或等于 node.val 的所有节点的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键 小于 节点键的节点。

节点的右子树仅包含键 大于 节点键的节点。

左右子树也必须是二叉搜索树。

示例：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]

输出：[30,36,21,36,35,26,15,33,8]

可使用以下main函数：

#include <iostream>

#include <queue>

#include <stack>

#include<cstdlib>

#include <cstring>

#include<map>

using namespace std;

struct TreeNode

{

    int val;

    TreeNode *left;

    TreeNode *right;

    TreeNode() : val(0), left(NULL), right(NULL) {}

    TreeNode(int x) : val(x), left(NULL), right(NULL) {}

    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}

};

TreeNode* inputTree()

{

    int n,count=0;

    char item[100];

    cin>>n;

    if (n==0)

        return NULL;

    cin>>item;

    TreeNode* root = new TreeNode(atoi(item));

    count++;

    queue<TreeNode*> nodeQueue;

    nodeQueue.push(root);

    while (count<n)

    {

        TreeNode* node = nodeQueue.front();

        nodeQueue.pop();

        cin>>item;

        count++;

        if (strcmp(item,"null")!=0)

        {

            int leftNumber = atoi(item);

            node->left = new TreeNode(leftNumber);

            nodeQueue.push(node->left);

        }

        if (count==n)

            break;

        cin>>item;

        count++;

        if (strcmp(item,"null")!=0)

        {

            int rightNumber = atoi(item);

            node->right = new TreeNode(rightNumber);

            nodeQueue.push(node->right);

        }

    }

    return root;

}

void printTree(TreeNode* root) {

    if (root == NULL) {

      return;

    }

    bool isFirst=true;

    queue<TreeNode*> q;

    q.push(root);

    while(!q.empty()) {

        TreeNode* node = q.front();

        q.pop();

        if (node == NULL) {

          continue;

        }

        if (!isFirst)

            cout<<",";

        cout<<node->val;

        isFirst=false;

        q.push(node->left);

        q.push(node->right);

    }

}

int main()

{

    TreeNode* root;

    root=inputTree();

    TreeNode* res=Solution().bstToGst(root);

    printTree(res);

}

输入说明

首先输入结点的数目n（注意，这里的结点包括题中的null空结点）

然后输入n个结点的数据，需要填充为空的结点，输入null。

输出说明

输出节点的信息，以逗号分隔

输入范例

15
4 1 6 0 2 5 7 null null null 3 null null null 8

输出范例

30,36,21,36,35,26,15,33,8

实现思路

先对该树进行一次遍历得到所有元素的和sum，

然后对该树进行一次中序遍历，每一次都用sum减去该结点的前驱结点的值（该前驱结点是中序遍历的前驱），同时更新sum值。

实现代码

#include <iostream>

#include <queue>

#include <stack>

#include<cstdlib>

#include <cstring>

#include<map>

using namespace std;

struct TreeNode

{

    int val;

    TreeNode *left;

    TreeNode *right;

    TreeNode() : val(0), left(NULL), right(NULL) {}

    TreeNode(int x) : val(x), left(NULL), right(NULL) {}

    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}

};

TreeNode* inputTree()

{

    int n,count=0;

    char item[100];

    cin>>n;

    if (n==0)

        return NULL;

    cin>>item;

    TreeNode* root = new TreeNode(atoi(item));

    count++;

    queue<TreeNode*> nodeQueue;

    nodeQueue.push(root);

    while (count<n)

    {

        TreeNode* node = nodeQueue.front();

        nodeQueue.pop();

        cin>>item;

        count++;

        if (strcmp(item,"null")!=0)

        {

            int leftNumber = atoi(item);

            node->left = new TreeNode(leftNumber);

            nodeQueue.push(node->left);

        }

        if (count==n)

            break;

        cin>>item;

        count++;

        if (strcmp(item,"null")!=0)

        {

            int rightNumber = atoi(item);

            node->right = new TreeNode(rightNumber);

            nodeQueue.push(node->right);

        }

    }

    return root;

}

void printTree(TreeNode* root) {

    if (root == NULL) {

      return;

    }

    bool isFirst=true;

    queue<TreeNode*> q;

    q.push(root);

    while(!q.empty()) {

        TreeNode* node = q.front();

        q.pop();

        if (node == NULL) {

          continue;

        }

        if (!isFirst)

            cout<<",";

        cout<<node->val;

        isFirst=false;

        q.push(node->left);

        q.push(node->right);

    }

}

class Solution {
public:
    int pre = 0;
    int sum = 0;

    void taversal(TreeNode* root){
        if(root==NULL) return;
        taversal(root->left);
        sum+=root->val;
        taversal(root->right);
    }

    void fun(TreeNode* root){
        if(root==NULL) return;
        fun(root->left);
        int tem = root->val;//存当前结点的值，因为下一步操作会修改root的value
        root->val=sum-pre;
        sum-=pre;//更新sum
        pre=tem;//更新pre
        fun(root->right);
    }

    TreeNode* bstToGst(TreeNode* root) {
        taversal(root);
        //cout<<sum
        fun(root);
        return root;
    }
};

int main()

{

    TreeNode* root;

    root=inputTree();

    TreeNode* res=Solution().bstToGst(root);

    printTree(res);

}