在 Java 中实现树结构的常用方法是使用递归定义节点的类型,并使用指针或引用来连接节点之间的关系。
例如,可以定义一个名为 TreeNode
的类,其中包含一个数据域和两个指针(左右儿子):
class TreeNode {
int data;
TreeNode left;
TreeNode right;
public TreeNode(int data) {
this.data = data;
}
}
然后,可以使用这个类来创建树节点,并使用指针来连接它们:
TreeNode root= new TreeNode(1);
root.left = new TreeNode(2);
root.right = new TreeNode(3);
root.left.left = new TreeNode(4);
这样,就可以使用递归函数遍历整棵树,例如先序遍历:
void preOrder(TreeNode node) {
if (node == null) {
return;
}
System.out.print(node.data + " ");
preOrder(node.left);
preOrder(node.right);
}
也可以使用非递归的方式遍历树,例如使用栈来实现中序遍历:
void inOrder(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
TreeNode current = root;
while (current != null || !stack.isEmpty()) {
while (current != null) {
stack.push(current);
current = current.left;
}
current = stack.pop();
System.out.print(current.data + " ");
current = current.right;
}
}