树结构实例

public class TreeDemo {
    private List<Node> list = new ArrayList<>(); // 所有子节点

    class Node {                 // 节点
        String data;             // 数据
        String parent;            // 父级节点
    }

    private void add(String parent, String child) {
        Node node = new Node();    //新建Node对象
        node.data = child;        
        node.parent = parent;
        list.add(node);
    }

    private void findParent(String child) {
        for (int i = 0; i < list.size(); i++) {            //循环遍历集合
            if (list.get(i).data.equals(child)) {        //判断是否包含信息
                System.out.println("父节点是==》" + list.get(i).parent);//输出信息
            }
        }
    }

    private ArrayList<String> findChild(String parent) {
        ArrayList<String> childs = new ArrayList<String>();
        for (int i = 0; i < list.size(); i++) {            //循环遍历集合
            if (list.get(i).parent.equals(parent)) {     //判断是否包含信息
                childs.add(list.get(i).data);            //添加信息到集合
            }
        }
        for (String string : childs) {                    //遍历信息
            System.out.println(string +"\t");
        }
        return childs;
    }
    
    public static void main(String[] args) {
        TreeDemo demo = new TreeDemo();
        demo.add("班级", "1班");
        demo.add("班级", "2班");
        demo.add("班级", "3班");
        demo.add("1班", "小黑1");
        demo.add("1班", "小黑2");
        demo.add("1班", "小黑3");
        demo.add("2班", "小白1");
        demo.add("2班", "小白2");
        demo.add("2班", "小白3");
        demo.add("3班", "小黄1");
        demo.add("3班", "小黄2");
        demo.add("3班", "小黄3");
        demo.add("3班", "小黄4");
        demo.findParent("小黑1");
        demo.findChild("3班");
    }
}