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

最新推荐文章于 2023-07-29 11:34:50 发布

原创最新推荐文章于 2023-07-29 11:34:50 发布 · 685 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

剑_牛

当前文章被收录于：

剑_牛

29 篇文章 1 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍如何将一棵二叉搜索树转换为排序的双向链表，通过中序遍历思想，调整树中节点指针指向，实现转换过程。无需创建新节点，重点在于递归处理左右子树。

题目描述

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

思路

本题和leetcode14有相似之处。
采用树的中序历遍的思想。

解答

/*
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(NULL==pRootOfTree) return pRootOfTree;
        if(pRootOfTree->left==NULL && pRootOfTree->right==NULL) return pRootOfTree;
        TreeNode* left=Convert(pRootOfTree->left);
        TreeNode* p=left;
        while(p && p->right) p=p->right;
        if(left)
        {
            p->right=pRootOfTree;
            pRootOfTree->left=p;
        }
        TreeNode* right=Convert(pRootOfTree->right);
        if(right)
        {
            pRootOfTree->right=right;
            right->left=pRootOfTree;
        }
        return left? left:pRootOfTree;
    }
};