找到最大级别1的WBS并将所有级别2的WBS放入数组的简单方法？

# 假设WBS以树的形式存储，每个节点具有标识符和子节点列表的属性

def find_level_2_wbs(root_node):
    level_2_wbs = []  # 存储级别2的WBS的数组

    # 遍历最大级别1的WBS的子节点
    for child_node in root_node.children:
        if child_node.level == 2:  # 判断级别是否为2
            level_2_wbs.append(child_node)  # 将级别为2的WBS添加到数组中

    return level_2_wbs

# 示例用法
root_wbs = find_root_wbs()  # 获取最大级别1的WBS根节点
level_2_wbs = find_level_2_wbs(root_wbs)  # 找到所有级别2的WBS

# 输出结果
for wbs in level_2_wbs:
    print(wbs.identifier)