如何打印一棵树(Java)

时间:2022-08-26 20:54:35

1、有一棵多叉树,将它打印出来。

import java.util.LinkedList;

/**
* 需求:按层打印一棵树
* 说明:树是保存在一个链表中
* created by wangjunfu on 2017-05-25.
*/
public class TreeNode {
String data;
TreeNode parent;
LinkedList
<TreeNode> childlist;

TreeNode() {
data
= null;
childlist
= new LinkedList();
parent
= null;
}

//递归显示并打印一棵树
private static void displayTree(TreeNode f, int level) {

String preStr
= ""; // 打印前缀
for (int i = 0; i < level; i++) {
preStr
+= " ";
}

for (int i = 0; i < f.childlist.size(); i++) {
TreeNode t
= f.childlist.get(i);
System.out.println(preStr
+ "-" + t.data);

if (!t.childlist.isEmpty()) {
displayTree(t, level
+ 1);
}
}
}
}

 

2、按层打印一棵树

下面抛个砖,一棵树,如何按层去打印呢?

我使用了一个最笨的办法,就是先存在一个list里,然后再排序打印,我知道这不是我想要的结果,但我确实想不出其它方法了,希望看到的大神抽空回个帖子,万分感谢!

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.LinkedList;

/**
* 需求:按层打印一棵树
* 说明:树是保存在一个链表中
* created by wangjunfu on 2017-05-25.
*/
public class TreeNode_Level {
Integer level;
String data;
TreeNode_Level parent;
LinkedList
<TreeNode_Level> childlist;

public Integer getLevel() {
return level;
}

public void setLevel(Integer level) {
this.level = level;
}

TreeNode_Level() {
data
= null;
childlist
= new LinkedList();
parent
= null;
}

// 递归读取一棵树
private ArrayList<TreeNode_Level> displayTree(TreeNode_Level f, int level) {

ArrayList
<TreeNode_Level> result = new ArrayList<>();
if (f == null) {
return result;
}

for (int i = 0; i < f.childlist.size(); i++) {
TreeNode_Level obj
= f.childlist.get(i);
obj.level
= level;
result.add(obj);

if (!obj.childlist.isEmpty()) {
displayTree(obj, level
+ 1);
}
}

return result;
}

// 排序、按层打印结果
private void print() {
TreeNode_Level f
= null; // 到库里取数据

ArrayList
<TreeNode_Level> list = displayTree(f, level = 1);

Collections.sort(list,
new Comparator<TreeNode_Level>() {
@Override
public int compare(TreeNode_Level o1, TreeNode_Level o2) {
// 正序排序
return o1.getLevel().compareTo(o2.getLevel());
}
});

list.forEach(item
-> System.out.println(item.data));
}
}