【LeetCode】（二叉树）单值二叉树

单值二叉树

题目描述

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

解题思路

1. 用一个结点指针p指向root结点。
2. 如果p结点有左子树，就用p的data值与左子树的data值进行比对是否相等，然后递归访问左子树，用一个bool类型的左树标记来接收左树的情况
3. 如果p结点有右子树，就用p的data值与右子树的data值进行比对是否相等，然后递归访问右子树。用一个bool类型的右树标记来接收右树的情况
4. 最后返回左树标记与右树标记的&&运算结果

参考源码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
bool isUnivalTree(struct TreeNode* root){
    if(root == NULL)
        return false;
    if(root->left == NULL && root->right == NULL)
        return true;
        
    struct TreeNode* p = root;
    bool flagRight = true;
    bool flagLeft = true;
    
    if(p->left != NULL)
    {
        int leftValue = p->left->val;
        int rootValue = p->val;
        flagLeft =  (rootValue == leftValue) && isUnivalTree(p->left);
    }
    
    if(root->right != NULL)
    {
        int rightValue = p->right->val;
        int rootValue = p->val;
        flagRight =  (rootValue == rightValue) && isUnivalTree(p->right);
    }
    
    return flagRight && flagLeft;
}

复杂度分析

假设结点个数为n
时间复杂度：O(n)
空间复杂度：一次递归含最多需要：2个bool类型 + 4个int类型 + 1个结点类型，递归次数由结点个数决定，所以空间复杂度为O(n)