二叉搜索树与双向链表

题目描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。

要求不能创建任何新的结点，只能调整树中结点指针的指向。

注意：

• 需要返回双向链表最左侧的节点

样例

输入如下二叉搜索树，返回结果如下：

解题思路

• 描述

  很明显，这道题需要使用到递归，但是如果构造递归函数则是一个比较麻烦的事情。

  根据样例可以得知，调整点在于左子树、根节点、右子树。对于根节点需要将其与左子树最右的节点连接起来，需将其与右子树最左的节点连接起来。因此一种可行思路就是，传参根节点，返回两个参数，当前子树所形成链表的最左节点和最右节点。

• 实现代码：

  /*
  struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
  };
  */
  TreeNode* convert(TreeNode* root) {
      if(!root) // 如果根节点为空，直接返回NULL
      {
          return NULL;
      }
      pair<TreeNode* ,TreeNode*> res = dfs(root);
      return res.first;
  }
  
  // 返回值，一个是该子树形成链表最左边的节点，一个是该子树形成链表最右边的节点
  pair<TreeNode* ,TreeNode*> dfs(TreeNode* root)
  {
      if(!root->left && !root->right) // 无左右子树，返回该节点
      {
          return {root,root};
      }
      if(root->left && root->right) // 左右子树均在
      {
          pair<TreeNode* ,TreeNode*> left = dfs(root->left);
          pair<TreeNode* ,TreeNode*> right = dfs(root->right);
          left.second->right = root; // 这里是调整的部分
          root->left = left.second;
          root->right = right.first;
          right.first->left = root;
          return {left.first,right.second};
      }
      if(root->left) // 仅含左子树
      {
          pair<TreeNode* ,TreeNode*> left = dfs(root->left);
          left.second->right = root;
          root->left = left.second;
          return {left.first,root};
      }
      if(root->right) // 仅含右子树
      {
          pair<TreeNode* ,TreeNode*> right = dfs(root->right);
          root->right = right.first;
          right.first->left = root;
          return {root,right.second};
      }
  }