设计模式代码实战-组合模式-编程知识

设计模式代码实战-组合模式

1、问题描述

小明所在的公司内部有多个部门，每个部门下可能有不同的子部门或者员工。
请你设计一个组合模式来管理这些部门和员工，实现对公司组织结构的统一操作。部门和员工都具有一个通用的接口，可以获取他们的名称以及展示公司组织结构。

输入示例
MyCompany
8
D HR
E HRManager
D Finance
E AccountantA
E AccountantB
D IT
E DeveloperA
E DeveloperB

输出示例
Company Structure:
MyCompany
HR
HRManager
Finance
AccountantA
AccountantB
IT
DeveloperA
DeveloperB

2、组合模式

它将对象组合成树状结构来表示“部分-整体”的层次关系。组合模式使得客户端可
以统⼀处理单个对象和对象的组合，⽽⽆需区分它们的具体类型。

类似树形结构，能够通过树形结构获取树中所有节点的信息。所有节点都有实现相同的接口
请添加图片描述

3、代码

import com.sun.java.accessibility.util.TopLevelWindowListener;import java.util.*;public class Main{public static void main(String[] args){Scanner sc=new Scanner(System.in);String companyName=sc.nextLine();int number=Integer.parseInt(sc.nextLine());Company company=new Company(companyName);Department department=null;for(int i=0;i<number;i++){String s=sc.nextLine();String[] s1=s.split(" ");String a=s1[0];String b=s1[1];if(a.equals("D")){department=new Department(b);company.addDepartment(department);}else if(a.equals("E")){department.addEmployee(new Employee(b));}}company.Display(0);}
}interface Component{void Display(int depth);
}class Employee implements Component{private String name;public Employee(String name){this.name=name;}@Overridepublic void Display(int depth) {for(int i=0;i<depth;i++){System.out.print("  ");}System.out.println(this.name);}
}class Department implements  Component{List<Department> subDepartment;List<Employee> employees;String departmentName;public  Department(String name){subDepartment=new ArrayList<>();employees=new ArrayList<>();departmentName=name;}public void addEmployee(Employee e){employees.add(e);}public void addSubDepartment(Department d){subDepartment.add(d);}@Overridepublic void Display(int depth) {for(int i=0;i<depth;i++){System.out.print("  ");}System.out.println(""+this.departmentName);if(!subDepartment.isEmpty()){for(int i=0;i<subDepartment.size();i++){subDepartment.get(i).Display(depth+1);}}if(!employees.isEmpty()){for(int i=0;i<employees.size();i++){employees.get(i).Display(depth+1);}}}
}class Company implements Component{private String name;List<Department> departments;public Company(String name){this.name=name;departments=new ArrayList<>();}@Overridepublic void Display(int depth) {System.out.println("Company Structure:");System.out.println(this.name);if(!departments.isEmpty()){for(int i=0;i<departments.size();i++){departments.get(i).Display(depth+1);}}}public void addDepartment(Department d){departments.add(d);}
}