【LeetCode】965. 单值二叉树

题目大意

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

数据范围：

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

思路

取根节点的值，然后开始dfs深度优先遍历该树；
若某节点为空，包括根节点为空的情况，则返回true；
若某节点的值不等于根节点的值，则返回false；
若某节点的值等于根节点的值，则继续搜索其左右子节点，返回其左右子节点的逻辑与。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isUnivalTree(TreeNode* root) {
        return dfs(root, root->val);
    }
    bool dfs(TreeNode* root, int val){
        if(!root) return true;
        if(root->val != val) return false;
        return dfs(root->left, val) && dfs(root->right, val);
    }
};

复杂度

时间复杂度：每个节点都被遍历一次，则其时间复杂度为$O(n)$。