[LeetCode] 988. Smallest String Starting From Leaf (Daily Question)
SeangG
2024. 4. 17. 10:51
문제
You are given the root of a binary tree where each node has a value in the range [0, 25] representing the letters 'a' to 'z'.
Return the lexicographically smallest string that starts at a leaf of this tree and ends at the root.
As a reminder, any shorter prefix of a string is lexicographically smaller. - For example, "ab" is lexicographically smaller than "aba".
A leaf of a node is a node that has no children.
root(트리)가 주어졌을 때 (잎노드 -> 루트노드)로 가면서 만든 문자열 중 사전적으로 가장 작은 문자열을 찾는 문제이다.
Example 1:
Input: root = [0,1,2,3,4,3,4] Output: "dba"
Example 2:
Input: root = [25,1,3,1,3,0,2] Output: "adz"
Example 3:
Input: root = [2,2,1,null,1,0,null,0] Output: "abc"
풀이
아이디어
DFS로 잎노드까지 가는 모든 경로를 탐색하여 가장 작은 문자열을 찾는다.
코드
# 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:
# 알파벳 테이블
alpha = [chr(i) for i in range(ord('a'), ord('z')+1)]
def smallestFromLeaf(self, root: Optional[TreeNode], st="") -> str:
# 현재 노드가 존재하지 않는 경우
if root is None: return "z"*10000
# 문자열 추가
st = self.alpha[root.val] + st
# 잎노드인 경우 문자열 반환
if root.left is None and root.right is None:
return st
# 자식 노드 탐색
st1 = self.smallestFromLeaf(root.left, st)
st2 = self.smallestFromLeaf(root.right, st)
return min([st1, st2])
