Java中高效关联父子列表数据：从O(NM)到O(N+M)的优化实践

public class Product {
    private long productId;
    private String productName;
    private BigDecimal costAmount;
    private List<ProductSub> productSublist; // 用于存储子项列表

    // 构造函数、Getter和Setter（此处省略）
    public Product(long productId, String productName, BigDecimal costAmount) {
        this.productId = productId;
        this.productName = productName;
        this.costAmount = costAmount;
        this.productSublist = new ArrayList<>();
    }

    public long getProductId() { return productId; }
    public void setProductSublist(List<ProductSub> productSublist) { this.productSublist = productSublist; }
    public List<ProductSub> getProductSublist() { return productSublist; }
}

public class ProductSub {
    private long productId; // 与Product关联的ID
    private long productSubId;
    private String lineItemName;
    private BigDecimal subCost;

    // 构造函数、Getter和Setter（此处省略）
    public ProductSub(long productId, long productSubId, String lineItemName, BigDecimal subCost) {
        this.productId = productId;
        this.productSubId = productSubId;
        this.lineItemName = lineItemName;
        this.subCost = subCost;
    }

    public long getProductId() { return productId; }
}

// 假设 productList 和 productSubList 已从服务获取
List<Product> productList = new ArrayList<>(); // ... (填充数据)
List<ProductSub> productSubList = new ArrayList<>(); // ... (填充数据)

for (Product productItem : productList){
    // 对于每个产品，遍历并过滤整个产品子项列表
    List<ProductSub> productSubItems = productSubList.stream()
       .filter(x -> x.getProductId() == productItem.getProductId())
       .collect(Collectors.toList());
    productItem.setProductSubList(productSubItems);
}

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.math.BigDecimal; // 确保导入BigDecimal

public class ProductAssociationOptimizer {

    public static void main(String[] args) {
        // 模拟数据
        List<Product> productList = new ArrayList<>();
        productList.add(new Product(1L, "Laptop", new BigDecimal("1200.00")));
        productList.add(new Product(2L, "Mouse", new BigDecimal("25.00")));
        productList.add(new Product(3L, "Keyboard", new BigDecimal("75.00")));

        List<ProductSub> productSubList = new ArrayList<>();
        productSubList.add(new ProductSub(1L, 101L, "CPU", new BigDecimal("500.00")));
        productSubList.add(new ProductSub(1L, 102L, "RAM", new BigDecimal("150.00")));
        productSubList.add(new ProductSub(2L, 201L, "Optical Sensor", new BigDecimal("10.00")));
        productSubList.add(new ProductSub(1L, 103L, "SSD", new BigDecimal("200.00")));
        productSubList.add(new ProductSub(3L, 301L, "Mechanical Switch", new BigDecimal("30.00")));
        productSubList.add(new ProductSub(2L, 202L, "Scroll Wheel", new BigDecimal("5.00")));
        productSubList.add(new ProductSub(4L, 401L, "Unmatched Sub", new BigDecimal("10.00"))); // 模拟一个没有对应产品ID的子项

        // 优化后的数据关联方法
        Map<Long, List<ProductSub>> productSubMap = new HashMap<>();

        // 第一次遍历：构建Map
        for (ProductSub productSub : productSubList) {
            // computeIfAbsent 是一个非常方便的方法
            // 如果键不存在，则创建一个新的ArrayList并放入Map，然后将当前productSub添加到该列表中
            // 如果键已存在，则直接获取对应的List并添加当前productSub
            productSubMap.computeIfAbsent(productSub.getProductId(), k -> new ArrayList<>()).add(productSub);
        }

        // 第二次遍历：将子列表设置到父产品中
        for (Product product : productList) {
            // 从Map中获取对应的子列表，O(1)操作
            List<ProductSub> subs = productSubMap.get(product.getProductId());
            // 注意处理subs可能为null的情况，如果某个产品没有对应的子项
            product.setProductSublist(subs != null ? subs : new ArrayList<>());
        }

        // 验证结果
        for (Product product : productList) {
            System.out.println("Product: " + product.getProductName() + " (ID: " + product.getProductId() + ")");
            if (product.getProductSublist().isEmpty()) {
                System.out.println("  No sub-items.");
            } else {
                for (ProductSub sub : product.getProductSublist()) {
                    System.out.println("  - Sub-item: " + sub.getLineItemName() + " (SubID: " + sub.getProductSubId() + ")");
                }
            }
        }
    }
}