Java实现：从菜单树中移除指定条件的子节点

原代码的核心问题 #

1. 未同步递归结果：递归调用removeChildMenu(aMenuItem.children)后，没有将返回的处理后子节点数组赋值回aMenuItem.children，导致子节点的修改无法同步到父节点。
2. 列表操作错误：Arrays.asList()返回的是固定大小的列表，直接调用removeIf会抛出UnsupportedOperationException；且原逻辑仅在子节点为null时才过滤当前节点，逻辑顺序完全错误。
3. 变量名不规范：存在大小写不一致（如Children应为children）、拼写错误（aMenuitem与aMenuItem），不符合Java编码规范。

正确实现代码 #

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class MenuProcessor {

    // 自定义移除条件：根据menuID和menuType判断是否移除
    private static boolean shouldRemove(MenuItemObj item) {
        // 示例条件：可根据实际需求修改，比如移除menuType为"HIDDEN"或menuID为"DEL"的节点
        return "HIDDEN".equals(item.getMenuType()) || "DEL".equals(item.getMenuID());
    }

    public static MenuItemObj[] removeChildMenu(MenuItemObj[] menuItemArr) {
        if (menuItemArr == null || menuItemArr.length == 0) {
            return new MenuItemObj[0];
        }

        // 将数组转为可修改的ArrayList，避免固定大小列表的操作异常
        List<MenuItemObj> processedList = new ArrayList<>(Arrays.asList(menuItemArr));

        // 先递归处理每个节点的子节点，确保子节点的移除操作完成
        for (MenuItemObj item : processedList) {
            if (item.getChildren() != null) {
                // 将递归处理后的子节点数组赋值回原节点，同步修改结果
                item.setChildren(removeChildMenu(item.getChildren()));
            }
        }

        // 过滤当前层级中符合移除条件的节点
        processedList = processedList.stream()
                .filter(item -> !shouldRemove(item))
                .collect(Collectors.toList());

        // 转回数组返回
        return processedList.toArray(new MenuItemObj[0]);
    }
}

// 注意：MenuItemObj需要添加getter和setter方法，否则无法访问私有属性
class MenuItemObj {
    private String menuID;
    private String menuType;
    private String menuDescription;
    private MenuItemObj[] children;

    // getter和setter方法
    public String getMenuID() { return menuID; }
    public void setMenuID(String menuID) { this.menuID = menuID; }
    public String getMenuType() { return menuType; }
    public void setMenuType(String menuType) { this.menuType = menuType; }
    public String getMenuDescription() { return menuDescription; }
    public void setMenuDescription(String menuDescription) { this.menuDescription = menuDescription; }
    public MenuItemObj[] getChildren() { return children; }
    public void setChildren(MenuItemObj[] children) { this.children = children; }
}

关键说明 #

1. 递归同步：处理完子节点后，必须将返回的新数组通过setChildren赋值回原节点，确保父节点能拿到修改后的子节点列表。
2. 列表可修改性：使用new ArrayList<>(Arrays.asList(menuItemArr))创建可修改的列表，避免固定大小列表的操作异常。
3. 处理顺序：先递归处理所有子节点，再过滤当前层级的节点，确保子节点的移除操作完成后，再判断当前节点是否保留。
4. 条件封装：将移除条件封装到shouldRemove方法中，便于后续修改和维护。

内容的提问来源于stack exchange，提问作者vjkumar