【Leetcode】156. Binary Tree Upside Down

记录算法题解

已于 2022-12-10 15:55:56 修改

阅读量341

点赞数

分类专栏： LC 树、分治与堆文章标签：二叉树 leetcode 算法 dfs c++

于 2021-08-08 02:32:16 首次发布

本文链接：https://blog.csdn.net/qq_46105170/article/details/119498902

版权

LC 树、分治与堆专栏收录该内容

270 篇文章

订阅专栏

这篇博客介绍了如何使用深度优先搜索（DFS）算法将二叉树的结构翻转，使得原来的左子树变成右子树，右子树变成左子树。文章通过给出详细的代码实现，解释了在递归过程中如何处理节点以完成翻转操作，并讨论了算法的时间复杂度为O(h)，其中h为树的高度。

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

题目地址：

https://leetcode.com/problems/binary-tree-upside-down/

给定一棵二叉树，要求对其从树根出发的左链做如下图的操作。返回新二叉树。
在这里插入图片描述

思路是DFS，在DFS的时候返回当前子树翻转之后的新树树根。如果树空则返回null；如果当前节点没有左孩子，则当前节点就是新树树根，返回之。否则先递归处理左子树，则得到新树树根。然后处理当前节点，其左孩子的左孩子是其右孩子，然后其左孩子的右孩子就是自己。最后把当前节点的左右孩子设为空，然后递归返回上层。代码如下：

class Solution {
 public:
  TreeNode* upsideDownBinaryTree(TreeNode* root) { 
    return dfs(root); 
  }

  TreeNode* dfs(TreeNode* p) {
    if (!p) return nullptr;
    if (!p->left) return p;
    TreeNode* res = dfs(p->left);
    p->left->left = p->right;
    p->left->right = p;
    p->left = p->right = nullptr;
    return res;
  }
};

时空复杂度 $O (h)$ 。