【前端er每日算法】二叉树-654最大二叉树-617合并二叉树-700二叉搜索树中的搜索-98验证二叉搜索树

130 阅读3分钟

题目一 654. 最大二叉树

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:

创建一个根节点,其值为 nums 中的最大值。 递归地在最大值 左边 的 子数组前缀上 构建左子树。 递归地在最大值 右边 的 子数组后缀上 构建右子树。 返回 nums 构建的 最大二叉树 。

思路

做了中序后续构造二叉树后,这道题就easy了,找最大值,切分数组,构造左右子数组,over😄

var constructMaximumBinaryTree = function(nums) {
    let maxIndex = 0;
    let max = 0;
    let len = nums.length;
    if (len === 0) {
        return null;
    }
    for (let i = 0; i < len; i++) {
        if (nums[i] > max) {
            max = nums[i];
            maxIndex = i;
        }
    }
    const root = new TreeNode(max);
    root.left = constructMaximumBinaryTree(nums.slice(0, maxIndex));
    root.right = constructMaximumBinaryTree(nums.slice(maxIndex + 1, len));
    return root;
};

题目二 617. 合并二叉树

给你两棵二叉树: root1 和 root2 。

想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。

思路

这题啊,磕磕绊绊了,写了一会,写出个毛。。好像知道该那么同时遍历两个树,结果还是写的稀里糊涂。来吧,正确思路:

  1. 如果root1为空,root2不为空,返回root2,反之返回root1
  2. 都不为空的情况下,在root1基础上修改,加上root2的值
  3. 谁是root1的左孩子呢?root1.left = 递归root1.left,root2.left
  4. 谁是root1的右孩子呢?root1.right = 递归root1.right,root2.right

一直好奇,老外写代码是啥感觉,就跟中文写一样,是不是很爽。

var mergeTrees = function(root1, root2) {
    if (!root1) {
        return root2;
    }
    if (!root2) {
        return root1;
    }
    root1.val += root2.val;
    root1.left = mergeTrees(root1.left, root2.left);
    root1.right = mergeTrees(root1.right, root2.right);
    return root1;
};

// 8.24 我为什么写出来的这样的代码?。。。
var mergeTrees = function(root1, root2) {
    const merge = (root1, root2, root1Pre, isLeft) => {
        if (root1 && root2) {
            root1.val = root1.val + root2.val;
            merge(root1.left, root2.left, root1, true);
            merge(root1.right, root2.right, root1, false);
        } else if (!root1 && root2) {
            if (root1Pre) {
                isLeft ? root1Pre.left = root2 : root1Pre.right = root2;
            } else {
                return root2;
            }
        }
        return root1;
    };
    root1 = merge(root1, root2, null);
    return root1;
};

题目三 700. 二叉搜索树中的搜索

思路

做了上面一道题,再做这个,我很开心,因为会做,分分钟出来,啥也不说了,上代码

var searchBST = function(root, val) {
    if (!root) {
        return null;
    }
    if (root.val === val) {
        return root;
    } else if (root.val < val) {
        return searchBST(root.right, val);
    } else {
        return searchBST(root.left, val);
    }
};

var searchBST = function(root, val) {
    while (root) {
        if (root.val === val) {
            return root;
        } else if (root.val < val) {
            root = root.right;
        } else {
            root = root.left;
        }
    }
    return null
}

题目四 98.验证二叉搜索树

给你一个二叉树的根节点 root ,判断其是否是一个有效的二叉搜索树。

有效 二叉搜索树定义如下:

节点的左子树只包含 小于 当前节点的数。 节点的右子树只包含 大于 当前节点的数。 所有左子树和右子树自身必须也是二叉搜索树。

思路

好说歹说,这道题做出来了,虽然曲折了点。中间调试还花了一会,每次做不出来的时候就在想,我这脑子不适合当程序员🤔

这个题就是中序遍历二叉树,并且记录上一个节点,比较当前节点和上一个节点的大小,如果当前节点小于上一个节点了,说明不符合二叉搜素树。

这里还要注意的就是这个是有返回值的,判断左子树拿到返回值,如果不符合,就返回false了,下面就不判断了,然后当前节点,然后返回右子树的结果。

var isValidBST = function(root) {
    let pre;
    const inorder = (root) => {
        if (!root) {
            return true;
        }
        const leftValid = inorder(root.left);
        if (!leftValid) {
            return false;
        }
        if (pre && root.val <= pre.val) {
            return false;
        }
        pre = root;
        const rightValid = inorder(root.right);
        return rightValid;
    }
    return inorder(root);
};

碎碎念

这两天节奏终于下来了,所以有时间刷题了,跟着学了回溯算法后,好像懵懵懂懂了,之前一道题老是要想半天结果还是做不出来,这次刷题改策略了,想一会想不出来直接看答案,有时候答案还得想好一会才想通呢。

上周拉的二叉树的题目疯狂补,刷刷刷,如果一直是这个节奏就好了,但愿如此,可是不太可能。但起码舒服了几天是不,但是从五一回来后,感冒没断,好不容易要好了,结果又流鼻涕了,快快好起来吧。😄

今天继续水,哈哈哈