树状结构显示之递归练习并说明写递归代码的技巧

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;

/**
 * 说明：不能使用多层循环的方式，因为需要能支持任意层。
 */
@SuppressWarnings("unused")
public class TreeViewPractice {

  /**
   * 练习一：打印所有顶层部门及其子孙部门的信息（名称） 提示：假设有一个 打印部门树 的信息 的方法
   * 
   * 要求打印如下效果：
   * 
   * <pre>
   * 市场部
   * 宣传部
   * 业务部
   * 业务一部
   * 业务二部
   * 开发部
   * 开发一部
   * 开发二部
   * </pre>
   */
  @Test
  public void printAllDepts_1() {
    List<Department> topList = findTopLevelDepartmentList();

    // 方式一
    // for (Department top : topList) {
    // showTree(top);
    // }

    // 方式二
    showTreeList(topList);
  }

  /**
   * 显示一颗部门树中所有节点的信息
   * 
   * @param top
   *            树的顶点(根节点)
   */
  public void showTree(Department top) {
    // 顶点
    System.out.println(top.getName());
    // 子树
    for (Department child : top.getChildren()) {
      showTree(child);
    }
  }

  /**
   * 显示多颗树的所有节点的信息
   * 
   * @param topList
   */
  public void showTreeList(Collection<Department> topList) {
    for (Department top : topList) {
      // 顶点
      System.out.println(top.getName());
      // 子树
      showTreeList(top.getChildren());
    }
  }

  /**
   * 练习二：打印所有顶层部门及其子孙部门的信息（名称），用不同的缩进表示层次（使用全角空格）。<br>
   * 子部门的名称前比上级部门多一个空格，最顶层部门的名字前没有空格。 提示：假设有一个打印部门集合中所有部门信息的方法
   * 
   * 要求打印如下效果：
   * 
   * <pre>
   * ┣市场部
   *    ┣宣传部
   *    ┣业务部
   *       ┣业务一部
   *       ┣业务二部
   * ┣开发部
   *    ┣开发一部
   *    ┣开发二部
   * </pre>
   */
  @Test
  public void printAllDepts_2() {
    List<Department> topList = findTopLevelDepartmentList();

    showTreeList_2(topList, "┣");
  }

  // 显示树
  public void showTreeList_2(Collection<Department> topList, String prefix) {
    for (Department top : topList) {
      // 顶点
      System.out.println(prefix + top.getName());
      // 子树
      showTreeList_2(top.getChildren(), "   " + prefix);
    }
  }

  /**
   * 结构如下：
   * 
   * <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;
  }

}