标签：right TreeNode 156 nullptr 节点 二叉树 root LeetCode left

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-upside-down

题目描述

给你一个二叉树的根节点 root ，请你将此二叉树上下翻转，并返回新的根节点。

你可以按下面的步骤翻转一棵二叉树：

原来的左子节点变成新的根节点
原来的根节点变成新的右子节点
原来的右子节点变成新的左子节点

 

 

上面的步骤逐层进行。题目数据保证每个右节点都有一个同级节点（即共享同一父节点的左节点）且不存在子节点。

 

示例 1：

 

 

输入：root = [1,2,3,4,5]
输出：[4,5,2,null,null,3,1]
示例 2：

输入：root = []
输出：[]
示例 3：

输入：root = [1]
输出：[1]
 

提示：

树中节点数目在范围 [0, 10] 内
1 <= Node.val <= 10
树中的每个右节点都有一个同级节点（即共享同一父节点的左节点）
树中的每个右节点都没有子节点

 

解题思路

经过观察发现仅仅是从下到上将父节点变成右结点，左结点变成父结点，右结点变成左结点，所以可以从下至上依次改变状态，可以使用栈来辅助寻找每一个结点，也可以使用前序遍历。

代码展示

栈辅助解法

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* upsideDownBinaryTree(TreeNode* root) {
        TreeNode* p = root;
        stack<TreeNode*> szStack;
        if(root)
        {
            while(p)
            {
                szStack.push(p);
                p = p->left;
            }
            root = szStack.top();
            while(!szStack.empty())
            {
                p = szStack.top();
                szStack.pop();
                if(szStack.empty())
                {
                    p->left = nullptr;
                    p->right = nullptr;
                }
                else
                {
                    TreeNode* q = szStack.top();
                    p->left = q->right;
                    p->right = q;
                }
            }
        }
        return root;

    }
};

前序遍历解法

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* mRoot;
    void dfs(TreeNode* root, TreeNode* fa)
    {
        if(!root) return;
        if(!root->left)
            mRoot = root;
        dfs(root->left, root);
        root->left = fa? fa->right : nullptr;
        root->right = fa;
    }

    TreeNode* upsideDownBinaryTree(TreeNode* root) {
        dfs(root,nullptr);
        return mRoot;

    }
};

 

运行结果

 

 

标签：right,TreeNode,156,nullptr,节点,二叉树,root,LeetCode,left
来源： https://www.cnblogs.com/zhangshihang/p/16078509.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。