给你一棵二叉搜索树,请你 按中序遍历 将其重新排列为一棵递增顺序搜索树,使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。
输入:root = [5,3,6,2,4,null,8,1,null,null,null,7,9]
输出:[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]
思路:
先对原树进行中序遍历,将遍历的结果保存下来,之后根据保存下来的结果,按照题目要求重新建树。
调bug原因:
对栈中元素进行删除时,s.pop()语句位置放的不对。
一开始将树的左节点放入栈中,忘记置空。
有空需要补一下指针的知识了。
/**
* 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:
TreeNode* increasingBST(TreeNode* root) {
vector<int> ans;
stack<TreeNode*> s;
s.push(root);
while(s.size())
{
TreeNode* tmp=s.top();
if(tmp->left !=NULL){
s.push(tmp->left);
tmp->left=NULL;
}
else{
ans.push_back(tmp->val);
s.pop();
if(tmp->right!=NULL){
s.push(tmp->right);
tmp->right=NULL;
}
}
}
TreeNode * re=new TreeNode(ans[0]);
TreeNode *be=re;
for(int i=1;i<ans.size();i++){
be->right=new TreeNode(ans[i]);
be=be->right;
}
return re;
}
};