//Component：它是一个抽象角色，为要组合的对象提供统一的接口。
public abstract class Component{
    public abstract void operation();
    //集成该抽象类，叶子节点不能拥有的方法如果调用抛出异常
    //枝节点需要，重写这些方法
    public void add(Component component) {throw new UnsupportedOperationException();}
    public void remove(Component component) {throw new UnsupportedOperationException();}
    public Component getChild(int i) {throw new UnsupportedOperationException();}
}

//Leaf:组合中表示子节点对象，叶子节点不能有子节点。
class ConcreteComponent1 extends Component {
    public void operation() {System.out.println("叶节点1 ");}
}
class ConcreteComponent2 extends Component {
    public void operation() {System.out.println("叶节点2");}
}
class ConcreteComponent3 extends Component {
    public void operation() {System.out.println("叶节点3");}
}

//Composite：定义有枝节点的行为，用来存储部件，实现在Component接口中的有关操作，如增加（Add）和删除（Remove）。
public class Composite extends Component{
    //保存子节点
    private ArrayList<Component> children;
    public Composite() {
        //初始化
        children = new ArrayList<Component>();
    }
    //遍历调用所有子节点方法
    public void operation() {
        for(Component child: children) {
            child.operation();
        }
    }
    //添加子节点
    @Override
    public void add(Component c) {children.add(c);}
    //移除子节点
    @Override
    public void remove(Component c) {children.remove(c);}
    //获取子节点
    @Override
    public Component getChild(int i) {return children.get(i);}
}

//Client：使用 Component 部件的对象。
public class Client{
    public static void main(String[] args){
        // 生成树根，并为其增加两个叶子节点
        Composite root = new Composite();
        root.add(new ConcreteComponent1());
        root.add(new ConcreteComponent2());
       //创建一个枝节点
        Composite branch = new Composite();
        // 为branch增加叶节点
        branch.add(new ConcreteComponent3());
        // 为根增加个枝节点
        root.add(branch);
        // 遍历整棵树
        root.operation();
    }
}