diff --git a/src/class11/Code03_EncodeNaryTreeToBinaryTree.java b/src/class11/Code03_EncodeNaryTreeToBinaryTree.java index aa9f2ab..f8515ec 100644 --- a/src/class11/Code03_EncodeNaryTreeToBinaryTree.java +++ b/src/class11/Code03_EncodeNaryTreeToBinaryTree.java @@ -58,7 +58,7 @@ public class Code03_EncodeNaryTreeToBinaryTree { cur.right = tNode; } cur = tNode; - cur.left = en(child.children); + cur.left = child.children != null ? en(child.children) : null; } return head; } @@ -72,7 +72,7 @@ public class Code03_EncodeNaryTreeToBinaryTree { } public List de(TreeNode root) { - List children = new ArrayList<>(); + List children = root != null ? new ArrayList<>() : null; while (root != null) { Node cur = new Node(root.val, de(root.left)); children.add(cur);