LeetCode --翻转二叉树

题目描述:

翻转一棵二叉树。
示例:
输入:
4
/ /
2 7
/ / / /
1 3 6 9
输出:
4
/ /
7 2
/ / / /
9 6 3 1

思路分析:

要想交换一个二叉树,只需要在遍历每个节点的时候,交换其左右两个孩子的位置,即可完成对整颗二叉树的逆置。
注意,此题不可以使用中序遍历来的反转二叉树。

递归法:

通过先序遍历每个节点,然后交换左右两个孩子,来实现对整个二叉树的翻转。
代码:

    TreeNode* invertTree(TreeNode* root) {
    
    
            if(root==NULL) return root;         //递归终止条件
            swap(root->left,root->right);       //中,
            invertTree(root->left);				//左
            invertTree(root->right);			//右
            return root;

DFS翻转二叉树

代码:

    TreeNode* invertTree(TreeNode* root) {
    
    
        if(root == NULL) return root;
        stack<TreeNode*> sta;           //深度遍历需要借助栈
        sta.push(root);                //先把头节点放到栈中去;
        while(!sta.empty())
        {
    
    
            TreeNode* node=sta.top();   //获取到栈顶节点,然后弹出
            sta.pop();               
            swap(node->left,node->right);        //交换栈顶节点左右孩子
            if(node->right)  sta.push(node->right);     //先压右孩子,因为栈是先进后出
            if(node->left)   sta.push(node->left);      //在压左孩子
        }
        return root;
    }

层序优先遍历反转二叉树

代码:

        TreeNode* invertTree(TreeNode* root) {
    
    
            queue<TreeNode*> que;              //层序遍历需要借助队列
            if(root!=NULL)  que.push(root);
            while(!que.empty())
            {
    
    
                int size=que.size();
                for(int i=0;i<size;i++)
                {
    
    
                    TreeNode* node=que.front();
                    que.pop();
                    //只要在遍历每个节点的时候,交换子节点的左右位置即可;
                    swap(node->left,node->right);
                    if(node->left) que.push(node->left);                //因为是队列,所以先左再右
                    if(node->right) que.push(node->right);
                }
            }
            return root;
        }

总结:

其实对于反转二叉树,最核心的思想便是遍历二叉树的每个节点,然后交换左右两个孩子的位置即可。

猜你喜欢

转载自blog.csdn.net/ALITAAAA/article/details/109367684