뒤죽박죽 코딩 이야기

2022년 04월 16일 토요일 - 블랙잭 너무 꿀잼이라구~~

오늘 올려볼 문제는 538번 Convert BST to Greater Tree 이라는 문제이다.

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

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

미디엄치고 너무 쉬워씀

입력

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

풀이 및 코드

주어진 Binaray Search Tree를 Greater Tree로 바꾸어 리턴하는 문제다.

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

dfs를 사용하는데 오른쪽 노드부터 탐색하면서 합을 구하고 노드의 값을 바꾸는 방식으로 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    int sum = 0;
    public TreeNode convertBST(TreeNode root) {
        solve(root);

        return root;
    }

    public void solve(TreeNode root)
    {
        if(root == null) return;

        solve(root.right);

        sum += root.val;

        root.val = sum;

        solve(root.left);
    }
}

제출 화면

오늘은 문제를 이해하는 시간이 문제를 푸는 시간보다 오래걸렸다.

내일 문제는 재밌는 문제가 나왔으면 좋겠다.

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