剑指offer 二叉搜索树与双向链表（C++）

题目描述

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

解题思路

中序遍历二叉搜索树可得到一个排序序列，将其放到vector容器中，通过数组下标的循环来修改每个结点指针的指向即可。

代码实现

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    vector<TreeNode*> InOrderV;
    void GetInOrderTree(TreeNode* pRootOfTree)// 犯低级错误 耽误时间
    {
        if(pRootOfTree)
        {
            if(pRootOfTree->left)
                GetInOrderTree(pRootOfTree->left);
            InOrderV.push_back(pRootOfTree);
            if(pRootOfTree->right)
                GetInOrderTree(pRootOfTree->right);
        }
    }
    
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree == NULL)
            return NULL;
        GetInOrderTree(pRootOfTree);
        if(InOrderV.size() == 1) //对数组中元素个数进行判断
            return InOrderV[0];
        int i = 0;
        for(i = 0; i < InOrderV.size(); i++)
        {
            if(i == 0)
            {
                InOrderV[i]->right = InOrderV[i + 1]; // 上面不判断的话 如果只有一个结点 这里出错
                continue;
            }
            if(i == InOrderV.size() - 1)
            {
                InOrderV[i]->left = InOrderV[i - 1];
                continue;
            }
            InOrderV[i]->left = InOrderV[i - 1];
            InOrderV[i]->right = InOrderV[i + 1];
        }
        return InOrderV[0];
    }
};

没有定义额外的函数解答此题，要深入研究。

class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if (pRootOfTree == NULL)return NULL;
        TreeNode *pointer = NULL;
        convert2List(pRootOfTree, pointer);
        while (pointer->left!=NULL)
        {
            pointer = pointer->left;
        }
        return pointer;
    }
    void convert2List(TreeNode* pRoot,TreeNode *&pointer)
    {
        if (pRoot == NULL)
        {
            return;
        }
        {
            if (pRoot->left != NULL)
            {
                convert2List(pRoot->left,pointer);
            }
 
            pRoot->left = pointer;
            if (pointer != NULL)
            {
                pointer->right = pRoot;
            }
             
            pointer = pRoot;
            if (pRoot->right!=NULL)
            {
                convert2List(pRoot->right, pointer);
            }
        }
    }
};

总结反思

1. 犯低级错误是一个很要命的事情。

2. 任何一个程序都要选好运行时指针或数组下标是否越界问题。