将一个二叉查找树按照中序遍历转换成双向链表。
样例
给定一个二叉查找树:
4
/ \
2 5
/ \
1 3
返回 1<->2<->3<->4<->5
。
思路:把二叉查找树的值按照中序遍历的顺序存到数组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;
}
};