牛客网-二叉搜索树与双向链表

牛客网-二叉搜索树与双向链表

题目描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

解题思路： 此题是《剑指offer》书上的题，基本的思路是，我们附设一个last指针，指向将树转换成双向链表后的最后一个节点。递归地将左子树变成链表，然后通过修改last和root的左右指针，将root收录到左子树的链表中，然后last指向root节点，最后将右子树转换成单向链表。

LeetCode上也有一道这题的《剑指offer》原题，但是相同思路，代码无法AC，真是奇怪。

牛客题：二叉搜索树与双向链表
LeetCode面试题36. 二叉搜索树与双向链表

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if (!pRootOfTree) return NULL;
        TreeNode* last = NULL;
        helper(pRootOfTree, last);
        TreeNode* head = last;
        while (head->left) {
            head = head->left;
        }
        return head;
    }
private:
    void helper(TreeNode* root, TreeNode* &last) {
        if (!root) return;
        helper(root->left, last);
        if (last) {
            last->right = root;
        }
        root->left = last;
        last = root;
        helper(root->right, last);
    }
};