在Java中,要實現TreeNode類的節點復制,你可以創建一個新的TreeNode對象,并將原始節點的值、子節點等信息復制到新對象中。以下是一個簡單的示例:
class TreeNode {
int val;
List<TreeNode> children;
public TreeNode(int val) {
this.val = val;
this.children = new ArrayList<>();
}
}
public class TreeNodeCopy {
public static void main(String[] args) {
// 創建一個簡單的樹結構
TreeNode root = new TreeNode(1);
TreeNode child1 = new TreeNode(2);
TreeNode child2 = new TreeNode(3);
TreeNode child3 = new TreeNode(4);
root.children.add(child1);
root.children.add(child2);
child1.children.add(child3);
// 復制樹結構
TreeNode copiedRoot = copyTreeNode(root);
// 輸出復制后的樹結構
System.out.println("Original tree:");
printTree(root, 0);
System.out.println("\nCopied tree:");
printTree(copiedRoot, 0);
}
public static TreeNode copyTreeNode(TreeNode node) {
if (node == null) {
return null;
}
// 創建一個新的TreeNode對象,并將原始節點的值復制到新對象中
TreeNode copiedNode = new TreeNode(node.val);
// 遍歷原始節點的子節點,并將它們復制到新對象中
for (TreeNode child : node.children) {
copiedNode.children.add(copyTreeNode(child));
}
return copiedNode;
}
public static void printTree(TreeNode node, int level) {
if (node == null) {
return;
}
// 打印縮進
for (int i = 0; i < level; i++) {
System.out.print(" ");
}
// 打印節點值
System.out.println(node.val);
// 遞歸打印子節點
for (TreeNode child : node.children) {
printTree(child, level + 1);
}
}
}
這個示例中,我們首先創建了一個簡單的樹結構,然后使用copyTreeNode
方法復制了這個樹結構。copyTreeNode
方法遞歸地遍歷原始節點的子節點,并將它們復制到新對象中。最后,我們使用printTree
方法輸出原始樹結構和復制后的樹結構,以便進行比較。