剑指OFFER 二叉搜索树与双向链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
分析
先按中序遍历一次,然后根据中序遍历的顺序来组织双向链表
代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<TreeNode*> v;
void construct_list(void)
{
int size = v.size();
if(size == 1){
v[0]->left = NULL;
v[0]->right = NULL;
return;
}
v[0]->left = NULL;
for(int i=1;i<size;i++)
{
v[i-1]->right = v[i];
v[i]->left = v[i-1];
}
v[size-1]->right = NULL;
}
//中序遍历
void recur(TreeNode* node)
{
if(node == NULL)return;
recur(node->left);
v.push_back(node);
recur(node->right);
}
TreeNode* Convert(TreeNode* root)
{
if(root == NULL)return NULL;
recur(root);
construct_list();
return v[0];
}
};