[剑指offer] 二叉搜索树和双向链表的转化

题目:二叉搜索树和双向链表的转化

  • 二叉搜索树里面每个节点有两个指针,分别指向left 和 right。
  • 双向链表内的每个节点也有两个指针,分别指向left 和 right。
  • 所以,两者是可以相互转化的。
  • 我们知道二叉搜索树的中序序列有序的。所以我们基于中序序列来将其转化为双向链表。
  • 对于每一个节点来说,它的left是它中序遍历的上一个元素,right是中序遍历的下一个元素。
  • 所以在中序遍历的过程中,我们简历一个lastNode变量,让其维持中序遍历访问的上一个元素。
  • 则lastNode->right = current Node, currentNode->left = lastNode
  • 通过这样的方式可以将二叉搜素树转化为双向链表
  • 代码如下:
//
// Created by 梁栋 on 2019-03-31.
// 将一颗搜索二叉树转变成排序双向链表
//
#include <iostream>
#include <vector>
#include <string>
#include <cstring>
using namespace std;
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
class Solution {
public:
    void ConvertBase(TreeNode* root, TreeNode** lastNode){
        if(root == NULL)
            return NULL;
        if(root->left){
            ConvertBase(root->left, lastNode);
        }
        root->left = *lastNode;
        if(*lastNode){
            (*lastNode)->right = root;
        }
        (*lastNode) = root;
        if(root->right){
            ConvertBase(root->right, lastNode);
        }
    }
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree == NULL)
            return NULL;
        TreeNode* lastNode = NULL;
        ConvertBase(pRootOfTree, &lastNode);
        TreeNode* res = pRootOfTree;
        while (res && res->left){
            res = res->left;
        }
        return res;
    }
};


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章