[LeetCode] 1339번 문제를 풀어보았다. (ft. java)

2022년 12월 10일 토요일 - 와우 블로그 오랜만~~~~~

오늘 올려볼 문제는 1339번 Maximum Product of Splitted Binary Tree 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

맥에서 캡처하니 shottr 쓸 수 있는건 좋다 ㅎㅎ

입력

사진에서도 볼 수 있듯이 Binary Tree 하나가 입력으로 들어온다.

---

풀이 및 코드

주어진 트리를 2개의 서브트리로 분리하여 각 트리의 합을 곱했을 때 가장 큰 값이 되는 값을 구해서 10^9 + 7로 나눈 값을 리턴하는 문제이다.

오늘은 처음부터 정답을 생각해냈다.

dfs로 각 노드가 루트가 되는 서브트리가 되었을 때 그 서브트리의 합을 노드에다가 적어두고 전체 트리의 합을 구했다.

이후 bfs로 순회하면서 전체합에서 노드의 값을 뺀 값과 노드의 값을 곱해서 최대값을 찾는 방식으로 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    final int mod = 1000000007;
    long result = 0;
    int sum = 0;
    public int maxProduct(TreeNode root) {
        sum = setSum(root);
        
        var q = new LinkedList<TreeNode>();
        q.add(root);
        
        while(!q.isEmpty()) {
            var size = q.size();
            
            for(int i = 0; i < size; i++) {
                var now = q.poll();
                if(now == null) continue;
                
                result = Math.max(result, (long)now.val * (long)(sum - now.val));
                
                q.add(now.left);
                q.add(now.right);
            }
        }
        
        return (int)(result % mod);
    }
    
    public int setSum(TreeNode root) {
        if(root == null) return 0;
        
        root.val += setSum(root.left) + setSum(root.right);
        
        return root.val;
    }
}

---

제출 화면

이렇게 오랜만에 블로그를 쓰니깐 또 뭔가 느낌이 남다르다.

이제는 문서화를 열심히 하려고하다보니 집에 와서 할 게 더 늘은거 같지만 그래도 열심히 해야겠다.

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.