[LeetCode] 129번 Sum Root to Leaf Numbers 문제를 풀어보았다. (ft. java)

2023년 03월 14일 화요일 - 오랜만이구만

오늘 올려볼 문제는 129번 Sum Root to Leaf Numbers 이라는 문제이다.

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

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

왜 요즘 문제가 잘 안풀리지..

입력

사진에서도 볼 수 있듯이 Binaray Tree 1개가 입력으로 들어온다.

---

풀이 및 코드

root 노드부터 leaf 노드까지의 길을 한 숫자라고 봤을 때 해당 바이너리 트리에서 나오는 모든 숫자의 합을 구하는 문제이다.

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

dfs 사용해서 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    int result = 0;
    
    public int sumNumbers(TreeNode root) {
        if(root != null) solve(root, 0);

        return result;
    }
    
    private void solve(TreeNode root, int num) {
        if(root.left == null && root.right == null) {
            result += num * 10 + root.val;
            return;
        }
        
        if(root.left != null) solve(root.left, num * 10 + root.val);
        if(root.right != null) solve(root.right, num * 10 + root.val);
    }
}

---

제출 화면

요즘따라 쉬운문제도 잘 못 풀고 그런다...

더 분발해야지...

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