LeetCode 250. Count Univalue Subtrees

Given a binary tree, count the number of uni-value subtrees.

A Uni-value subtree means all nodes of the subtree have the same value.

For example:
Given binary tree,

          5
         / \
        1   5
       / \   \
      5   5   5

return 4.

思路：
1. univalue subtree是指子树的每个节点的值都相等。例子中三个leaf node是子树且节点值相等，另外还有

          5
           \
            5

2. 需要dfs遍历，提取信息，做出判断。提取的信息是左右子树是否univalue或NULL，以及univalue的值是多少。如果root的值等于左右子树的univalue，则总的子树个数+1；否则传递信息表明当前不是univalue subtree

int countUnivalSubtrees(TreeNode* root) {
    //
    int res=0;
    int value=root->val;
    helper(root,res,value);
    return res;
}

bool helper(TreeNode* root,int&res,int&value){
    //
    if(!root) return true;
    int lvalue=root->val, rvalue=root->val;
    if(helper(root->left,res,lvalue)&&helper(root->right,res,rvalue)){
        if(lvalue==rvalue&&lvalue==root->value){
            res+=1;
            value=root->val;
            return true;    
        }
    }
    return false;

}