遍历树节点

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>

* 市场部

* 宣传部

* 业务部

* 业务一部

* 业务二部

* 开发部

* 开发一部

* 开发二部

* </pre>

*/

@Test

public void printAllDepts_1() {

// 顶级树的根节点集合

List<Department> topList = findTopLevelDepartmentList();

// // 方式一

// for (Department department : topList) {

// showTreeInfo(department);

// }

// 方式二

showTreeListInfo(topList);

}

// 方式一： 显示一棵树的信息

private void showTreeInfo(Department top) {

// 顶点信息

System.out.println(top.getName());

// 子树

for (Department department : top.getChildren()) {

showTreeInfo(department);

}

}

// 方式二： 显示多棵树的信息

private void showTreeListInfo(Collection<Department> topList) {

for (Department department : topList) {

// 顶点信息

System.out.println(department.getName());

// 子树 注： 此处返回的是一个set集合，所有要将参数定义为Collection类型

showTreeListInfo(department.getChildren());

}

}

/**

* 练习二：打印所有顶层部门及其子孙部门的信息（名称），用不同的缩进表示层次（使用全角空格）。<br>

* 子部门的名称前比上级部门多一个空格，最顶层部门的名字前没有空格。 提示：假设有一个打印部门集合中所有部门信息的方法

*

* 要求打印如下效果：

*

* <pre>

* ┣市场部

* ┣宣传部

* ┣业务部

* ┣业务一部

* ┣业务二部

* ┣开发部

* ┣开发一部

* ┣开发二部

* </pre>

*/

@Test

public void printAllDepts_2() {

List<Department> topList = findTopLevelDepartmentList();

showTreeListInfo(topList, "");

}

private void showTreeListInfo(Collection<Department> topList,String prefix) {

for (Department department : topList) {

// 根节点信息

System.out.println(prefix + "┣"+department.getName());

// 子树

showTreeListInfo(department.getChildren()," "+prefix);

}

}

// 实例应用

/**

* 遍历部门树，把所有的部门都改变名称后放到同一个list中返回，通过名称中的空格表示层次

* @param topList

* @param removedDepartment 要移除的分支部门： 该部门和该部门的子部门都不要

* @return

*/

public static List<Department> getAllDepartmentList(List<Department> topList,Department removedDepartment) {

List<Department> list = new ArrayList<Department>();

walkTree(topList,"┣",list,removedDepartment);

return list;

}

// 递归遍历

private static void walkTree(Collection<Department> topList,String prefix,List<Department> list,Department removedDepartment){

for (Department top : topList) {

// 去掉指定的部门分支

if(removedDepartment!=null && top.getId().equals(removedDepartment.getId())){

continue;

}

// 根节点 注： 不要修改Session缓存中的对象，最好使用副本(重新new一个对象即可)

Department copy = new Department();

copy.setId(top.getId());

copy.setName(prefix + top.getName());

list.add(copy);
// 注： 添加的是copy对象,只需id和name属性即可

// 子树

walkTree(top.getChildren(), "　　" + prefix,list,removedDepartment); // 此处使用的是全角的空格，避免HTML忽略多余的空格

}

}

/**

* 结构如下：

*

* <pre>

* ┣市场部

* ┣宣传部

* ┣业务部

* ┣业务一部

* ┣业务二部

* ┣开发部

* ┣开发一部

* ┣开发二部

* </pre>

*

* @return 所有最顶层的部门的列表

*/

public static List<Department> findTopLevelDepartmentList() {

Department dept_1_1 = new Department();

dept_1_1.setId(new Long(11));

dept_1_1.setName("宣传部");

Department dept_1_2 = new Department();

dept_1_2.setId(new Long(12));

dept_1_2.setName("业务部");

Department dept_1_2_1 = new Department();

dept_1_2_1.setId(new Long(121));

dept_1_2_1.setName("业务一部");

Department dept_1_2_2 = new Department();

dept_1_2_2.setId(new Long(122));

dept_1_2_2.setName("业务二部");

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("开发一部");

Department dept_2_2 = new Department();

dept_2_2.setId((new Long(22)));

dept_2_2.setName("开发二部");

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;

}

}