【Leetcode】897. Increasing Order Search Tree

原创已于 2024-10-22 15:20:38 修改 · 120 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表 #leetcode #二叉树 #数据结构 #c++

于 2020-04-06 11:41:11 首次发布

LC 树、分治与堆专栏收录该内容

270 篇文章

订阅专栏

本文介绍两种方法将二叉搜索树转换为单调递增的链表，第一种方法通过分治策略，先将左右子树转换为链表，再连接；第二种方法采用中序遍历，直接构建链表。第一种方法的时间复杂度为O(n^2)，空间复杂度为O(h)，第二种方法的时空复杂度均为O(n)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

https://leetcode.com/problems/increasing-order-search-tree/

给定一棵二叉搜索树，将其flatten成一个单调增加的链表，链表的next规定为TreeNode的right。

法1：分治。开一个函数，可以将一棵BST铺平成一个链表，并且返回链表的头尾；如果链表为空则返回两个nullptr。先分治两边，接着处理当前节点，即将左边连到当前节点，并且将当前节点连到右边。代码如下：

/**
 * 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) { return dfs(root).first; }

  pair<TreeNode*, TreeNode*> dfs(TreeNode* root) {
    if (!root) return {nullptr, nullptr};
    TreeNode *l1, *r1, *l2, *r2;
    tie(l1, r1) = dfs(root->left);
    tie(l2, r2) = dfs(root->right);
    if (r1) r1->right = root;
    root->right = l2;
    root->left = nullptr;
    return {l1 ? l1 : root, r2 ? r2 : root};
  }
};

时间复杂度 $O (n)$ ，空间 $O (h)$ 。

法2：直接中序遍历即可。每次append一个节点到链表末尾。代码如下：

class Solution {
 public:
  TreeNode* increasingBST(TreeNode* root) {
    TreeNode *dummy = new TreeNode(0), *cur = dummy;
    dfs(root, cur);
    return dummy->right;
  }

  void dfs(TreeNode* p, TreeNode*& cur) {
    if (!p) return;
    dfs(p->left, cur);
    cur->right = p;
    cur = p;
    cur->left = nullptr;
    dfs(p->right, cur);
  }
};

时空复杂度一样。