[LintCode]378.将二叉查找树转换成双链表

将一个二叉查找树按照中序遍历转换成双向链表。

样例

给定一个二叉查找树:

    4
   / \
  2   5
 / \
1   3

返回 1<->2<->3<->4<->5

二叉查找树是满足以下条件的二叉树:1.左子树上的所有节点值均小于根节点值,2右子树上的所有节点值均不小于根节点值,3,左右子树也满足上述两个条件。

思路:把二叉查找树的值按照中序遍历的顺序存到数组vector中,再把数组中的值按顺序构造成双链表。

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 * Definition of Doubly-ListNode
 * class DoublyListNode {
 * public:
 *     int val;
 *     DoublyListNode *next, *prev;
 *     DoublyListNode(int val) {
 *         this->val = val;
           this->prev = this->next = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: The root of tree
     * @return: the head of doubly list node
     */
    DoublyListNode* bstToDoublyList(TreeNode* root) {
        if(root == NULL)
            return NULL;
            
        vector<int> list;
        treeToList(root, list);//先把值存到数组
        
        DoublyListNode* head = listToDlist(list);//数组构造双链表
        return head;
    }
    void treeToList(TreeNode *TreeRoot, vector<int> &list) {
        //树结点的值通过中序遍历存入数组list
        if(TreeRoot != NULL) {
            treeToList(TreeRoot->left, list);
            list.push_back(TreeRoot->val);
            treeToList(TreeRoot->right, list);
        }
    }
     //通过已知数组构造双链表
    DoublyListNode* listToDlist(vector<int> &list) {
        DoublyListNode* head=NULL,* p=NULL;
        head = p = new DoublyListNode(list[0]);//数组首位赋值
        for(int i=1; i<list.size(); i++) {     //从数组第二位开始搜索
            DoublyListNode* q = new DoublyListNode(list[i]);
            p->next = q;
            q->prev = p;
            p = q;//将q插入到链表尾部
        }
        return head;
    }
};



    原文作者:二叉查找树
    原文地址: https://blog.csdn.net/zwy1258432405/article/details/77350346
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞