LeetCode 每日一题——965. 单值二叉树

1.题目描述

965. 单值二叉树

如果二叉树每个节点都具有相同的值，那么该二叉树就是单值二叉树。

只有给定的树是单值二叉树时，才返回 true；否则返回 false。

示例 1：

输入：[1,1,1,1,1,null,1]
输出：true

示例 2：

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

2.思路

2.1 代码

题目比较简单，直接遍历二叉树，然后使用 pre 存放上一个节点的值，用当前节点值与上一个节点的值进行比较，如果值不同将ans 赋值为 false 即可。这里选用前序遍历，代码如下：

class Solution {
    int pre;
    boolean ans = true;

    public boolean isUnivalTree(TreeNode root) {
        pre = root.val;
        process(root);
        return ans;
    }

    public void process(TreeNode root) {
        if (root == null) {
            return;
        }
        if (root.val != pre) {
            ans = false;
        }
        process(root.left);
        process(root.right);
    }
}

2.2 测试结果

通过测试

3.总结

• 遍历二叉树使用当前节点与前一个节点值进行比较，如果发现不同将结果设置成 false 返回