Leetcode—1038.从二叉搜索树到更大和树【中等】

2023每日刷题（四十九）

Leetcode—1038.从二叉搜索树到更大和树

在这里插入图片描述

算法思想

二叉搜索树的中序遍历（左根右）结果是一个单调递增的有序序列，我们反序进行中序遍历（右根左），即可以得到一个单调递减的有序序列。通过累加单调递减的有序序列，我们可以得到大于等于 node.val 的新值，并重新赋值给 node。

按照“右根左”的顺序，递归遍历二叉搜索树，累加遍历到的所有节点值到 s 中，然后每次赋值给对应的 node 节点。

实现代码

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:int s = 0;void dfs(struct TreeNode* root) {if(root == nullptr) {return;}dfs(root->right);s += root->val;root->val = s;dfs(root->left);}TreeNode* bstToGst(TreeNode* root) {dfs(root);return root;}
};

运行结果

在这里插入图片描述
别问我为什么不给C语言实现代码，因为力扣编译器有问题

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     struct TreeNode *left;*     struct TreeNode *right;* };*/
int s = 0;void dfs(struct TreeNode* root) {if(root == NULL) {return;}dfs(root->right);s += root->val;root->val = s;dfs(root->left);
}struct TreeNode* bstToGst(struct TreeNode* root) {dfs(root);return root;
}