leetcode 单值二叉树

短命女 2022-03-20 08:26 130阅读 0赞

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

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

示例 1：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MjUzNTY5_size_16_color_FFFFFF_t_70]

输入：\[1,1,1,1,1,null,1\]  
输出：true  
示例 2：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MjUzNTY5_size_16_color_FFFFFF_t_70 1]

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

提示：

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

不多说。。。上代码

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

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MjUzNTY5_size_16_color_FFFFFF_t_70]: /images/20220320/5b725264730c47dabf7933f5cc42961b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MjUzNTY5_size_16_color_FFFFFF_t_70 1]: /images/20220320/d69dbc8588354edb98e38f52f2250bae.png