刷题:二叉搜索树与双向链表

  • 题目:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
  • 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);
    }
};

猜你喜欢

转载自blog.csdn.net/nickkissbaby_/article/details/89321868