Leetcode 250. Count Univalue Subtrees 谷歌面试, Java解法详解

最新推荐文章于 2022-08-04 11:45:33 发布

原创

最新推荐文章于 2022-08-04 11:45:33 发布 · 418 阅读

0 ·

CC 4.0 BY-SA版权

这篇博客详细解析了LeetCode 250题目的解决方案，针对谷歌面试中可能出现的同值子树问题，提供了两种Java解答思路。解法一的时间复杂度为O(N^2)，通过递归计算每个节点的子树是否为同值子树。解法二尚未展开，暗示可能有更优的O(N)时间复杂度解法。

这是一道谷歌会考的题目

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.

Example :

Input:  root = [5,1,5,5,5,null,5]

              5
             / \
            1   5
           / \   \
          5   5   5

Output: 4

解法一

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int countUnivalSubtrees(TreeNode root) {
        if(root==null) return 0;
        int totalCount = countUnivalSubtrees(root.left)+countUnivalSubtrees(root.right);
        if(isUni(root)) totalCount+=1;
        return totalCount;
    }
    
    public boolean isUni(TreeNode root){
        if(root==null) return true;
        if(root.left!=