Java项目中Pair数据存储优化：如何替代三维列表实现高效查询？

你的三维ArrayList<ArrayList<ArrayList<Pair>>>方案存在明显缺陷：维护成本高（需手动对应维度与gender/foodPreference的映射关系）、查询效率低（要遍历索引定位）、扩展性差（新增gender或foodPreference类别时需手动扩容多层列表），完全不适合按gender + foodPreference组合查询的场景。以下是更简洁高效的替代方案：

方案1：嵌套HashMap（直观易实现） #

用两层HashMap构建索引，外层键为gender，内层键为foodPreference，最终值为对应组合的Pair列表。查询时可直接通过键值定位，时间复杂度O(1)。

代码示例 #

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

// 构建索引
Map<String, Map<String, List<Pair>>> pairIndex = new HashMap<>();
for (Pair pair : pairList) {
    String gender = pair.getGender();
    String foodPref = pair.getFoodPreference();
    
    // 自动创建不存在的gender键对应的内层Map
    pairIndex.computeIfAbsent(gender, k -> new HashMap<>());
    // 自动创建不存在的foodPreference键对应的列表
    pairIndex.get(gender).computeIfAbsent(foodPref, k -> new ArrayList<>());
    // 将Pair加入对应分组
    pairIndex.get(gender).get(foodPref).add(pair);
}

// 查询示例：获取"male"且"veggie"的Pair列表
List<Pair> targetPairs = pairIndex.getOrDefault("male", Collections.emptyMap())
                                  .getOrDefault("veggie", Collections.emptyList());

方案2：自定义复合键（面向对象，扩展性强） #

如果后续可能扩展查询维度（比如加入年龄区间），可以创建一个包含gender和foodPreference的复合键类，重写equals()和hashCode()方法，用单层HashMap存储。

代码示例 #

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;

// 自定义复合键类
class PairQueryKey {
    private final String gender;
    private final String foodPreference;

    public PairQueryKey(String gender, String foodPreference) {
        this.gender = gender;
        this.foodPreference = foodPreference;
    }

    // 重写equals：两个字段完全匹配才视为相同键
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        PairQueryKey that = (PairQueryKey) o;
        return Objects.equals(gender, that.gender) &&
               Objects.equals(foodPreference, that.foodPreference);
    }

    // 重写hashCode：基于两个字段生成哈希值
    @Override
    public int hashCode() {
        return Objects.hash(gender, foodPreference);
    }
}

// 构建索引
Map<PairQueryKey, List<Pair>> pairIndex = new HashMap<>();
for (Pair pair : pairList) {
    PairQueryKey key = new PairQueryKey(pair.getGender(), pair.getFoodPreference());
    // 自动创建不存在的键对应的列表并添加Pair
    pairIndex.computeIfAbsent(key, k -> new ArrayList<>()).add(pair);
}

// 查询示例
PairQueryKey queryKey = new PairQueryKey("male", "veggie");
List<Pair> targetPairs = pairIndex.getOrDefault(queryKey, Collections.emptyList());

方案对比 #

• 嵌套HashMap：无需额外编写类，实现简单直观，适合当前固定的两个查询维度场景。
• 自定义复合键：结构更严谨，后续新增查询字段时只需修改复合键类，扩展性更强。

无论选择哪种方案，都比三维ArrayList更高效、更易维护，完全适配你的组合查询需求。

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