组合设计模式1

/**
 * 组合设计模式：组合设计模式主要是组装树状的数据，
 * 当数据拥有同类型的子节点时，可以使用组合设计模式对数据进行组装
 * @author 
 * @version 0.1
 */
public class TreeNode {
    private String name;
    private TreeNode parent;
    private List<TreeNode> children = new ArrayList<TreeNode>();

    public TreeNode(String name){
        this.name = name;
    }
    //添加子节点方法
    public void add(TreeNode node){
        children.add(node);
    }
    //删除子节点方法
    public void remove(TreeNode node){
        children.remove(node);
    }
    //返回子节点的迭代器
    public Iterator<TreeNode> getChildren(){
        return children.iterator();
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public TreeNode getParent() {
        return parent;
    }

    public void setParent(TreeNode parent) {
        this.parent = parent;
    }

}

/**
 * @author 
 * @version 0.1
 */
public class Tree {
    TreeNode root = null;

    public Tree(String name){
        root = new TreeNode(name);
    }

    public static void main(String[] args){
        Tree tree = new Tree("A");
        TreeNode nodeB = new TreeNode("B");
        TreeNode nodeC = new TreeNode("C");

        nodeB.add(nodeC);
        tree.root.add(nodeB);
    }
}