114. Flatten Binary Tree to Linked List

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

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

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种将二叉树扁平化的算法实现，通过两种不同的方法达到目的：一种是非递归的方式，另一种是更为简洁的递归方式。核心在于调整二叉树节点的左右子节点连接，使得最终每个节点只有一个子节点，从而形成一个链表。

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

class Solution {
public:
	void Core(TreeNode* root)
	{
		int flag = 0;
		if (root->left != NULL && (root->left->left != NULL || root->left->right != NULL))
			Core(root->left);
		if (root->right != NULL && (root->right->left != NULL || root->right->right != NULL))
			Core(root->right);
		TreeNode* tmp = root->right;
		if (root->left != NULL) {
			root->right = root->left;
			flag = 1;
		}
		root->left = NULL;
		while (root->right != NULL)
			root = root->right;
		if (flag)
			root->right = tmp;
	}
	void flatten(TreeNode* root) {
		if (root == NULL || (root->left == NULL && root->right == NULL))
			return;
		Core(root);
	}
};

今天头痛，就这样吧，非递归的比较简洁。

void flatten(TreeNode *root) {
	while (root) {
		if (root->left && root->right) {
			TreeNode* t = root->left;
			while (t->right)
				t = t->right;
			t->right = root->right;
		}

        if(root->left)
		    root->right = root->left;
		root->left = NULL;
		root = root->right;
	}
}