题目地址:
https://leetcode.com/problems/smallest-string-starting-from-leaf/
给定一棵二叉树,求其所有从叶子到树根的路径中字典序最小的那个。题目保证节点只含 0 ∼ 25 0\sim 25 0∼25的整数,并且路径要将数字转为英文小写字母, 0 0 0对应 a a a, 1 1 1对应 b b b以此类推。
思路是DFS。当DFS到叶子的时候,更新答案即可。回溯的时候需要多加小心。代码如下:
public class Solution {
private String res;
public String smallestFromLeaf(TreeNode root) {
res = "";
// 特判空树,以后DFS的时候只走非null节点
if (root == null) {
return res;
}
dfs(root, new StringBuilder());
return res;
}
private void dfs(TreeNode cur, StringBuilder sb) {
sb.append((char) ('a' + cur.val));
// 走到叶子了,可以更新答案
if (cur.left == null && cur.right == null) {
// 注意,这里的或者用|,而不用||,目的在于强行让sb翻转
if (res.isEmpty() | res.compareTo(sb.reverse().toString()) > 0) {
res = sb.toString();
}
// 回溯之前sb还需要翻转回去
sb.reverse();
return;
}
if (cur.left != null) {
dfs(cur.left, sb);
// 回溯的时候恢复现场
sb.setLength(sb.length() - 1);
}
if (cur.right != null) {
dfs(cur.right, sb);
sb.setLength(sb.length() - 1);
}
}
}
class TreeNode {
int val;
TreeNode left, right;
public TreeNode(int val) {
this.val = val;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( h ) O(h) O(h)。