03. 剑指offer刷题-二叉树篇（第二部分）

原创

于 2024-08-12 20:21:51 发布 · 203 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #C++ #数据结构

class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree) {
        
		if(pRootOfTree == nullptr) return nullptr;
		vector<TreeNode*> cur = traversal(pRootOfTree);
		return cur[0];
    }
    // 这道题需要用到「分解问题」的思维，想把整棵链表，可以先把左右子树变成链表
    //然后把 root.val 接在中间，这样就形成了整棵 BST 的环形链表
	vector<TreeNode*> traversal(TreeNode* pRootOfTree){

		if(pRootOfTree == nullptr) return {nullptr, nullptr};

		vector<TreeNode*> left =  traversal(pRootOfTree->left);
		vector<TreeNode*> right =  traversal(pRootOfTree->right);

		pRootOfTree->left = left[1];
		if(left[1] != nullptr) left[1]->right = pRootOfTree;

		pRootOfTree-&