题目描述:
将一个二叉查找树按照中序遍历转换成双向链表。
样例:
给定一个二叉查找树:
4
/ \ 2 5
/ \ 1 3
返回 1<->2<->3<->4<->5。
思路讲解:这道题主要想考察的点就是二叉树的遍历与双向链表的建立,所以我们只需要分别解决这两个问题就好了,首先我们利用递归将二叉树的中序列表找出,然后我们再递归建立链表,这里建立链表我们需要注意的点就是链表的首节点的前节点是空的,尾节点的后节点也是空的,所以我们只需要注意这两个问题就好了。
代码详解:
/** * 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) {
// Write your code here
if(root==NULL){
return NULL;
}
vector<int>res;
traveltreenode(root,res);
printint(res);
return buildlistnode(res,0,NULL);
}
void traveltreenode(TreeNode * head,vector<int>&res){//二叉树的中序遍历
if(head!=NULL){
traveltreenode(head->left,res);
res.push_back(head->val);
traveltreenode(head->right,res);
}
}
void printint(vector<int>res){
for(int i=0;i<res.size();i++){
cout<<res[i]<<" ";
}
cout<<endl;
}
DoublyListNode* buildlistnode(vector<int>res,int index,DoublyListNode* pre){//链表的建立
if(index==res.size()-1){//对于最后一个节点的特殊处理,也解决了递归的出口
DoublyListNode* end=new DoublyListNode();
end->val=res[index];
end->prev=pre;
end->next=NULL;
return end;
}
DoublyListNode* head=new DoublyListNode();
head->val=res[index];
head->prev=pre;
head->next=buildlistnode(res,index+1,head);
return head;
}
};