【ACWing】49. 二叉搜索树与双向链表

题目地址:

https://www.acwing.com/problem/content/description/87/

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
注意:
需要返回双向链表最左侧的节点。

数据范围:
树中节点数量 [ 0 , 500 ] [0,500] [0,500]

直接中序遍历一遍即可。代码如下:

#include <stack>
using namespace std;

struct TreeNode {
  int val;
  TreeNode *left, *right;
  TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
  TreeNode* convert(TreeNode* root) {
    stack<TreeNode*> stk;
    auto dummy = new TreeNode(0), prev = dummy;
    while (stk.size() || root) {
      while (root) {
        stk.push(root);
        root = root->left;
      }

      root = stk.top(); stk.pop();
      prev->right = root;
      root->left = prev;
      prev = prev->right;

      root = root->right;
    }

    root = dummy->right;
    // 把dummy和最终答案断开
    dummy->right = nullptr;
    if (root) root->left = nullptr;
    return root;
  }
};

时间复杂度 O ( n ) O(n) O(n),空间 O ( h ) O(h) O(h)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值