前端算法-二叉树翻转

62 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情

题目

给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。

输入: root = [4,2,7,1,3,6,9]
输出: [4,7,2,9,6,3,1]

思路一

我们可以使用递归进行实现,先判断当前形参root是否为null,如果是则直接返回null,否则就继续往下执行,我们在创建一个left变量和right变量,用于接受调用自身函数传值为root左节点和root右节点的返回值,然后我们将返回值重新赋值给形参root的左节点和形参root的右节点,最后返回root形参即可

/**
 * @param {TreeNode} root
 * @return {TreeNode}
 */
var invertTree = function(root) {
    if (root == null) {
        return null
    }
    let left = invertTree(root.left)
    let right = invertTree(root.right)
    root.left = right
    root.right = left
    return root
};

思路二

我们这里可以使用到深度优先搜索,用栈的另一种写法的后序遍历来实现,我们把要处理的节点放入栈后然后放入一个空指针作为标记,如果出栈元素为空,则说明该节点是需要处理的节点,我们先创建一个st变量,他是一个空栈,然后再判断当前的root是否不等于null,如果不是则将root使用push方法添加到st栈中,然后使用循环进行变量,循环条件为当前的st变量的长度不为0,在循环中我们声明一个node变量用于存储栈顶的元素,然后判断node是否为null,如果不是null,就不是需要处理的节点,然后按照中子节点,右子节点,左子节点的顺序使用push方法添加到st栈中,如果为空则删除空节点,然后再删除需要处理的节点,在交换其左右子树,最后将形参root返回出去即可

var invertTree = function(root) {
    var st = [];
    if (root !== null) st.push(root);
    while (st.length !== 0) {
        var node = st[st.length - 1];
        if (node !== null) {
            st.pop();
            st.push(node);                        
            st.push(null);
            if (node.right) st.push(node.right); 
            if (node.left) st.push(node.left); 
        } else {
            st.pop();
            node = st.pop();
            var temp = node.left;
            node.left = node.right;
            node.right = temp;
        }
    } 
    return root;
}