《剑指offer》二叉搜索树与双向链表

一、题目描述

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

二、输入描述

输入一棵二叉搜索树

三、输出描述

将该二叉搜索树转换成一个排序的双向链表

四、牛客网提供的框架

/*
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)
    {

    }
};

五、解题思路

按照中序遍历，按照左子树、根节点、右子树的顺序。

六、代码

#include<iostream>

using namespace std;


struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
class Solution {
public:
    TreeNode* newHead;
    TreeNode* currNode;

    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        newHead = NULL;
        currNode = NULL;
        midOrder(pRootOfTree);

        return newHead;
    }

    void midOrder(TreeNode* pRootOfTree)    //中序（递归）遍历
    {
        if(!pRootOfTree) return;
        if(pRootOfTree->left) midOrder(pRootOfTree->left);  //左子树

        //加入链表
        if(!newHead)   //是否是第一个节点
        {
            newHead = pRootOfTree;
            currNode = pRootOfTree;
        }
        else
        {
            currNode->right = pRootOfTree;
            pRootOfTree->left = currNode;
            currNode = pRootOfTree;
        }
        if(pRootOfTree->right) midOrder(pRootOfTree->right);    //右子树
    }

};