36.二叉搜索树与双向链表.md

36.二叉搜索树和双向链表

题目描述

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

思路 & 解答

主要是使用递归，设置一个head和end，head只在第一次设置值，后面就是将end的right指向当前node，将当前node的left指向end，把end指向当前的node。

代码如下：

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}

public class Solution {
    public TreeNode head = null;
    public TreeNode end = null;

    public TreeNode Convert(TreeNode pRootOfTree) {
        invert(pRootOfTree);
        return head;
    }

    public void invert(TreeNode node) {
        if (node == null) {
            return;
        }
        invert(node.left);
        if (end == null) {
            head = node;
            end =node;
        }else{
            end.right = node;
            node.left = end;
            end =node;
        }
        invert(node.right);
    }
}

C++ 代码如下：

class Solution {
public:
    TreeNode *head;
    TreeNode *end;

    TreeNode *Convert(TreeNode *pRootOfTree) {
        invert(pRootOfTree);
        return head;
    }

    void invert(TreeNode *node) {
        if (node == NULL) {
            return;
        }
        invert(node->left);
        if (end == NULL) {
            head = node;
            end = node;
        } else {
            end->right = node;
            node->left = end;
            end = node;
        }
        invert(node->right);
    }
};

时间复杂度：O(N)。 空间复杂度：O(N),主要是递归的栈空间。