每日力扣-树-单值二叉树

74 阅读1分钟

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

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

解题思路:没啥好说的,深度优先遍历

class Solution {
public:
    int flag = 1;
    bool isUnivalTree(TreeNode* root) {
        if(root == nullptr) return true;
        int ans = root->val;
        dfs(root,ans);
        if(flag == 1) return true;
        return false;
    }
    void dfs(TreeNode* root,int &ans)
    {
        if(root == nullptr) return;
        if(root->val != ans)
        {
            flag = -1;
        }
        dfs(root->left,ans);
        dfs(root->right,ans);
    }
};