- 题目:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
- idea:由于中序遍历可以得到一个从小到大的序列,因此选择中序遍历二叉搜索树,并用一个变量prev保存前一个遍历的节点。注意Helper函数中传入的prev指针为引用,因此整个遍历过程修改的prev都是同一个变量prev。在遍历的时候,应该有root->left=prev, prev->right=root。然后当前节点应该被更新为prev,即prev=root。最后,返回的节点应该是最左端的节点。即假如当前节点为叶子节点或者没有左子树即返回当前节点,否则返回root。
- 注意指针一定要初始化,初始化为NULL也是初始化
- code
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
TreeNode* Helper(TreeNode* root, TreeNode* &prev){
if (!root) return NULL;
TreeNode *left = NULL, *right = NULL;
if (root->left)
left = Helper(root->left, prev);
root->left = prev;
if (prev) prev->right = root;
prev = root;
if (root->right)
right = Helper(root->right, prev);
if (!root->left) return root;
else return left;
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if (!pRootOfTree) return NULL;
TreeNode *prev = NULL;
return Helper(pRootOfTree, prev);
}
};