[코테] 리트코드 트리 687. Longest Univalue Path

687. Longest Univalue Path

문제 링크

https://leetcode.com/problems/longest-univalue-path/description/

문제 설명

Given the root of a binary tree, return the length of the longest path, where each node in the path has the same value. This path may or may not pass through the root.

The length of the path between two nodes is represented by the number of edges between them.

제한 사항

• The number of nodes in the tree is in the range [0, 10^4].
• -1000 <= Node.val <= 1000
• The depth of the tree will not exceed 1000.

입출력 예 #1

1. Input : root = [5,4,5,1,1,null,5]
2. Output : 2
3. Explanation: The shown image shows that the longest path of the same value (i.e. 5).

입출력 예 #2

1. Input : root = [1,4,5,4,4,null,5]
2. Output : 2
3. Explanation : The shown image shows that the longest path of the same value (i.e. 4).

문제 풀이

bfs 기법을 이용해서 해당 노드의 자식 노드들을 queue에 넣고, queue에 남아있는 항목이 있으면 depth에 1씩 더하면서 반복한다.

class Solution:
    result: int = 0

    def longestUnivaluePath(self, root: Optional[TreeNode]) -> int:
        def dfs(node: TreeNode):
            if node is None:
                return 0

            # 존재하지 않는 노드까지 DFS 재귀 탐색
            left = dfs(node.left)
            right = dfs(node.right)

            # 현재 노드가 자식 노드와 동일한 경우 거리 1 증가
            if node.left and node.left.val == node.val :
                left += 1
            else : 
                left = 0

            if node.right and node.right.val == node.val :
                right += 1
            else : 
                right = 0 

            # 왼쪽과 오른쪽 자식 노드 간 거리의 합 최댓값이 결과
            self.result = max(self.result, left + right)
            # 자식 노드 상태값 중 큰 값 리턴
            return max(left, right)

        dfs(root)
        return self.result