牛客每日刷题之二叉树

最新推荐文章于 2024-08-28 17:14:56 发布

_18shou

最新推荐文章于 2024-08-28 17:14:56 发布

阅读量767

点赞数 8

分类专栏：基础算法 java 文章标签：面试职场和发展

本文链接：https://blog.csdn.net/m0_60264772/article/details/126360155

版权

基础同时被 3 个专栏收录

19 篇文章 0 订阅

订阅专栏

java

18 篇文章 0 订阅

订阅专栏

算法

17 篇文章 0 订阅

订阅专栏

✅作者简介：我是18shou，一名即将秋招的java实习生

✨个人主页：_18shou

🔥系列专栏：牛客刷题专栏

📃推荐一款模拟面试、刷题神器👉 在线刷题面经模拟面试

题目

操作给定的二叉树,将其变换为源二叉树的镜像。
数据范围:二叉树的节点数0≤n≤1000，二叉树每个节点的值0≤ val≤1000
要求:空间复杂度0(m).本题也有原地操作,即空间复杂度0(1)的解法,时间复杂度0(n)

思路

主要是利用栈(或队列)遍历树的所有节点node，并交换每个node的左/右子节点
算法流程:
1、特例处理:当pRoot为空时，直接返回null ;
2、初始化:栈(或队列)，本文用栈stack，并加入根节点pRoot。
3、循环交换:当栈 stack为空时跳出;
3.1、出栈:记为node ;
3.2、添加子节点:将node左和右子节点入栈; .
3.3、交换:交换node的左1右子节点。
4、返回值:返回根节点pRoot。

题解

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return TreeNode类
     */
    public TreeNode Mirror (TreeNode pRoot) {
        // write code here
        if(pRoot == null) return null;
        // 构建辅助栈
        Stack<TreeNode> stack = new Stack<>();
        // 根节点入栈
        stack.add(pRoot);
        while(!stack.isEmpty()) {
            // 节点出栈
            TreeNode node = stack.pop();
            // 根节点的左右子树入栈
            if(node.left != null) stack.add(node.left);
            if(node.right != null) stack.add(node.right);
            // 左右子树交换
            TreeNode tmp = node.left;
            node.left = node.right;
            node.right = tmp;
        }
        return pRoot;
    }
}