首页 > 其他 > 详细

二叉搜索树转换成单链表

时间:2020-02-26 14:55:09      阅读:120      评论:0      收藏:0      [点我收藏+]

题目概述

二叉树数据结构TreeNode可用来表示单向链表(其中left置空,right为下一个链表节点)。实现一个方法,把二叉搜索树转换为单向链表,要求值的顺序保持不变,转换操作应是原址的,也就是在原始的二叉搜索树上直接修改。

思路

BST的半线性结构转换成线性结构,其实就是中序遍历而已。

/**
 * 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* convertBiNode(TreeNode* root) {
      if (root == nullptr) return root;
      std::stack<TreeNode*> stack;  // 中序遍历需要的栈
      while(root) {
        stack.push(root);
        root = root->left;
      }
      TreeNode* newHead = stack.top();  // 最左的那棵叶节点即为单链表头结点
      while(!stack.empty()) {
        TreeNode* leftNode = stack.top(); stack.pop();
        TreeNode* node = leftNode->right;
        while(node) {
          stack.push(node);
          node = node->left;
        }
        leftNode->left = nullptr;
        leftNode->right = stack.empty() ? nullptr : stack.top();
      }
      return newHead;
    }
};

二叉搜索树转换成单链表

原文:https://www.cnblogs.com/codemeta-2020/p/12366985.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!