package cn.itcast.oa.test;
import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;
import org.junit.Test;
import cn.itcast.oa.domain.Department;
/**
* 说明:不能使用多层循环的方式,因为需要能支持任意层。
*/
public class TreeViewPractice {
/**
* 练习一:打印所有顶层部门及其子孙部门的信息(名称) 提示:假设有一个 打印部门树 的信息 的方法
*
* 要求打印如下效果:
*
* <pre>
* 市场部
* 市场1部
* 市场2部
* 21
* 22
* 开发部
* 开发1部
* 开发2部
* </pre>
*/
@Test
public void printAllDepts_1() {
List<Department> topList = findTopLevelDepartmentList();
// // 方式一:
for (Department top : topList) {
printTree(top);
}
// 方式二:
//printTreeList(topList);
}
// 方式一:显示一颗以top为顶点的部门树的信息(显示所有的节点名称)
private void printTree(Department top) {
// 1个顶点
System.out.println(top.getName());
// 0或多个子树
for (Department child : top.getChildren()) {
printTree(child);
}
}
// 方式二:显示一堆树的信息
private void printTreeList(Collection<Department> topList) {
for (Department top : topList) {
// 1个顶点
System.out.println(top.getName());
// 0或多个子树
printTreeList(top.getChildren());
}
}
/**
* 练习二:打印所有顶层部门及其子孙部门的信息(名称),用不同的缩进表示层次(使用全角空格)。<br>
* 子部门的名称前比上级部门多一个空格,最顶层部门的名字前没有空格。 提示:假设有一个打印部门集合中所有部门信息的方法
*
* 要求打印如下效果:
*
* <pre>
* ┠市场部
* ┠市场1部
* ┠市场2部
* ┠21
* ┠22
* ┠开发部
* ┠开发1部
* ┠开发2部
* </pre>
*/
@Test
public void printAllDepts_2() {
List<Department> topList = findTopLevelDepartmentList();
printTreeList_2(topList, "┣");
}
// 显示一堆树的信息,并且名称前有空格
private void printTreeList_2(Collection<Department> topList, String prefix) {
for (Department top : topList) {
// 1个顶点
System.out.println(prefix + top.getName());
// 0或多个子树
printTreeList_2(top.getChildren(), " " + prefix);
}
}
/**
* 结构如下:
*
* <pre>
* ┠市场部
* ┠市场1部
* ┠市场2部
* ┠21
* ┠22
* ┠开发部
* ┠开发1部
* ┠开发2部
* </pre>
*
* @return 所有最顶层的部门的列表
*/
public static List<Department> findTopLevelDepartmentList() {
Department dept_1_1 = new Department();
dept_1_1.setId(new Long(11));
dept_1_1.setName("市场1部");
Department dept_1_2 = new Department();
dept_1_2.setId(new Long(12));
dept_1_2.setName("市场2部");
Department dept_1_2_1 = new Department();
dept_1_2_1.setId(new Long(121));
dept_1_2_1.setName("21");
Department dept_1_2_2 = new Department();
dept_1_2_2.setId(new Long(122));
dept_1_2_2.setName("22");
dept_1_2_1.setParent(dept_1_2);
dept_1_2_2.setParent(dept_1_2);
Set<Department> children_0 = new LinkedHashSet<Department>();
children_0.add(dept_1_2_1);
children_0.add(dept_1_2_2);
dept_1_2.setChildren(children_0);
// ================================
Department dept_1 = new Department();
dept_1.setId(new Long(1));
dept_1.setName("市场部");
dept_1_1.setParent(dept_1);
dept_1_2.setParent(dept_1);
Set<Department> children_1 = new LinkedHashSet<Department>();
children_1.add(dept_1_1);
children_1.add(dept_1_2);
dept_1.setChildren(children_1);
// ---
Department dept_2_1 = new Department();
dept_2_1.setId(new Long(21));
dept_2_1.setName("开发1部");
Department dept_2_2 = new Department();
dept_2_2.setId((new Long(22)));
dept_2_2.setName("开发2部");
Department dept_2 = new Department();
dept_2.setId(new Long(2));
dept_2.setName("开发部");
dept_2_1.setParent(dept_2);
dept_2_2.setParent(dept_2);
Set<Department> children_2 = new LinkedHashSet<Department>();
children_2.add(dept_2_1);
children_2.add(dept_2_2);
dept_2.setChildren(children_2);
// ---
List<Department> depts = new ArrayList<Department>();
depts.add(dept_1);
depts.add(dept_2);
return depts;
}
}
打印树状结构
最新推荐文章于 2023-03-16 23:00:00 发布