Flatten Binary Tree to Linked List

最新推荐文章于 2024-08-25 16:11:27 发布

原创最新推荐文章于 2024-08-25 16:11:27 发布 · 456 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

leetcode 专栏收录该内容

154 篇文章

订阅专栏

本文介绍了一种将二叉树结构通过先序遍历转换为单链表的算法实现。该方法首先进行先序遍历记录节点，然后重新连接这些节点形成扁平化的结构。整个过程不使用额外的数据结构，确保了空间复杂度为O(1)。

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

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
   void flatten(TreeNode *root) {
    	if(!root)	return;
    	vector<TreeNode*> preorder;
    	preorderTraversal(root,preorder);
    	size_t i=0;
    	for(;i!=(preorder.size()-1);i++)
    	{
    		preorder[i]->left=0;
    		preorder[i]->right=preorder[i+1];
    	}
    	preorder[i]->left=preorder[i]->right=0;
    }
    void preorderTraversal(TreeNode *root,vector<TreeNode*> &preorder)
    {
    	if(!root)	return;
    	preorder.push_back(root);
    	preorderTraversal(root->left,preorder);
    	preorderTraversal(root->right,preorder);
    }
};