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

最新推荐文章于 2021-10-21 10:02:51 发布

原创最新推荐文章于 2021-10-21 10:02:51 发布 · 322 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

43 篇文章

订阅专栏

题目要求在1秒时间和32768K空间限制下，将输入的二叉搜索树转换成排序的双向链表，且不能创建新结点，仅调整树中结点指针指向，并给出了相关代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

时间限制：1秒空间限制：32768K
题目描述
输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。
代码：

/*
*修改当前结点与上一节点的指针指向
*顺序是中序遍历的左根右的顺序
*/

C++:
/*
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 == nullptr)
            return nullptr;
        stack<TreeNode*> st;
        TreeNode* p = pRootOfTree;
        TreeNode* pre = nullptr;
        bool flag = true;//记录第一个
        while(p != nullptr || !st.empty()){
            while(p != nullptr){
                st.push(p);
                p = p->left;
            }
            p = st.top();
            st.pop();
            if(flag){//对第一个进行单独操作
                pRootOfTree = p;
                pre = pRootOfTree;
                flag = false;
            }
            else{
                pre->right = p;
                p->left = pre;
                pre = p;
            }
            p = p->right;
        }
        return pRootOfTree;
    }
};