965. 单值二叉树

965. 单值二叉树

如果二叉树每个节点都具有相同的值，那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时，才返回 true；否则返回 false。

示例 1：

输入：[1,1,1,1,1,null,1] 输出：true
示例 2：

输入：[2,2,2,5,2] 输出：false

提示：

1. 给定树的节点数范围是 [1, 100]。
2. 每个节点的值都是整数，范围为 [0, 99] 。

思路

按题意模拟即可，flag 可以提前剪枝

解答

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
var isUnivalTree = function (root) {
  let flag = true;
  const target = root.val;
  const visit = (node) => {
    if (!flag || !node) return false;
    if (node.val !== target) {
      flag = false;
      return false;
    }
    if (flag) visit(node.left);
    if (flag) visit(node.right);
  };
  visit(root);
  return flag;
};