LeetCode每日一题 538.把二叉搜索树转换为累加树,二叉树遍历顺序问题
题目描述
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
解题思路
排序树的先序遍历就是从小到大遍历,如果翻过来便是从大到小了,也就是先遍历右子树然后遍历根最后遍历左子树。在反序遍历的过程中需要记录已遍历过的结点的和,然后加到后续遍历的节点上。
题解代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
void houxu(TreeNode* node,int &sum){
if(node==NULL){
return;
}
if(node->right!=NULL)
houxu(node->right,sum);
sum+=node->val;
node->val=sum;
if(node->left!=NULL)
houxu(node->left,sum);
}
TreeNode* convertBST(TreeNode* root) {
int sum=0;
houxu(root,sum);
return root;
}
};
非递归方法
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* convertBST(TreeNode* root) {
stack<TreeNode*> nodeStack;
TreeNode* cur=root;
int sum=0;
while(cur!=NULL||!nodeStack.empty()){
if(cur!=NULL){
nodeStack.push(cur);
cur=cur->right;
}else{
cur=nodeStack.top();
nodeStack.pop();
sum+=cur->val;
cur->val=sum;
cur=cur->left;
}
}
return root;
}
};