[劍指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;
    }
};


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