(刷题笔记) Leetcode 897. 递增顺序查找树

最新推荐文章于 2021-04-16 19:30:22 发布

原创最新推荐文章于 2021-04-16 19:30:22 发布 · 337 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树

刷题笔记专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种通过中序遍历将给定的二叉搜索树转换为递增顺序搜索树的方法，使得树的最左侧节点成为根节点，每个节点仅有一个右子节点，展示了使用C++实现的具体代码，包括迭代和递归两种解决方案。

题目

给你一个树，请你按中序遍历重新排列树，使树中最左边的结点现在是树的根，并且每个结点没有左子结点，只有一个右子结点。

示例：

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

      5
     / \
    3    6
   / \    \
  2   4    8
 /        / \ 
1        7   9

输出：[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]

1
 \
  2
    \
     3
       \
        4
          \
           5
            \
             6
              \
               7
                \
                 8
                  \
                   9

提示：

给定树中的结点数介于 1 和 100 之间。
每个结点都有一个从 0 到 1000 范围内的唯一整数值。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/increasing-order-search-tree

解题思路

这道题与面试题 17.12. BiNode是一模一样的。
不过隔了一段时间又换了一种说法之后，我就只记得我写过，但是不记得怎么写的，然后就顺着写了一个中序遍历，但是没办法返回到头结点，然后就回来看了一下这个答案。

代码(C++)

/**
 * 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* increasingBST(TreeNode* root) {

        stack<TreeNode*> st;

        TreeNode* temp=root;
        TreeNode* now=NULL;
       
        while(!st.empty()||temp!=NULL){

            while(temp){
                st.push(temp);
                temp=temp->right;
            }

            temp=st.top();
            st.pop();
           
            temp->right=now;
            now=temp;
            temp=temp->left;
            now->left=NULL;
            
        } 

        return now;

        
        
    }
};

递归

/**
 * 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* now=NULL;
    void dfs(TreeNode* root){

        if(root==NULL) return;
        dfs(root->right);
        root->right=now;
        now=root;
        dfs(root->left);
        root->left=NULL;
    }
    TreeNode* increasingBST(TreeNode* root) {

        dfs(root);
        return now;
        
    }
};