하위 트리의 평균과 같은 개수의 노드

이진 트리의 root가 주어지면 노드 값이 하위 트리 값의 평균과 같은 노드 수를 반환합니다.

메모:

n 요소의 평균은 n 요소의 합계를 n로 나눈 다음 가장 가까운 정수로 내림한 것입니다.

root의 하위 트리는 root 및 모든 하위 항목으로 구성된 트리입니다.

예 1:



입력: 루트 = [4,8,5,0,1,null,6]
출력: 5
설명:
값이 4인 노드의 경우 하위 트리의 평균은 (4 + 8 + 5 + 0 + 1 + 6)/6 = 24/6 = 4입니다.
값이 5인 노드의 경우 하위 트리의 평균은 (5 + 6)/2 = 11/2 = 5입니다.
값이 0인 노드의 경우: 하위 트리의 평균은 0/1 = 0입니다.
값이 1인 노드의 경우: 하위 트리의 평균은 1/1 = 1입니다.
값이 6인 노드의 경우: 하위 트리의 평균은 6/1 = 6입니다.

예 2:



입력: 루트 = [1]
출력: 1
설명: 값이 1인 노드의 경우: 하위 트리의 평균은 1/1 = 1입니다.

제약:

트리의 노드 수가 [1, 1000] 범위에 있습니다.

0 <= Node.val <= 1000

해결책:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def sumcount(self, root):
        total = 0
        n = 0
        if root:
            ltotal, ln = self.sumcount(root.left)
            rtotal, rn = self.sumcount(root.right)
            total = ltotal + rtotal + root.val
            n = ln + rn + 1
            if root.val == total // n:
                self.ctr += 1
        return (total, n)

    def averageOfSubtree(self, root: Optional[TreeNode]) -> int:
        self.ctr = 0
        self.sumcount(root)
        return self.ctr