HashMap 中的部分搜索

Question

我需要创建电话簿之类的东西。它包含名称和编号。现在当我输入字母匹配列表时应该返回。对于下面给出的示例，当我键入 H 时，应返回包含 Harmer、Harris、Hawken、Hosler 的列表。当键入 Ha 时，应返回仅包含 Harmer、Harris、Hawken 的列表。

  Map<String, String> nameNum = new HashMap<String, String>();

  nameNum.put("Brown", "+1236389023");
  nameNum.put("Bob", "+1236389023");
  nameNum.put("Harmer", "+1236389023");
  nameNum.put("Harris", "+1236389023");
  nameNum.put("Hawken", "+1236389023");
  nameNum.put("Hosler", "+1236389023");

知道如何实现吗？ 提前致谢。

Answer 1

删除所有不包含关键部分的值：

yourMap.keySet().removeIf(key -> !key.contains(keyPart));

或正则表达式：

yourMap.keySet().removeIf(key -> !key.matches(".*keyPart.*"));

或过滤流并收集到新地图：

yourMap.entrySet().stream().filter(e -> e.getKey().contains(keyPart)).collect(Collectors.toMap(e -> e.getKey(), e -> e.getValue()));

Answer 2

是的，HashMap 不是正确的数据结构。正如 Bozho 所说，Trie 是正确的。

借助 Java 的板载工具，可以使用 TreeMap（或任何 SortedMap，实际上）：

public <V> SortedMap<String, V> filterPrefix(SortedMap<String,V> baseMap, String prefix) {
    if(prefix.length() > 0) {
        char nextLetter = prefix.charAt(prefix.length() -1) + 1;
        String end = prefix.substring(0, prefix.length()-1) + nextLetter;
        return baseMap.subMap(prefix, end);
    }
    return baseMap;
}

输出甚至可以按键排序。

这里是一个用法示例：

SortedMap<String, String> nameNum = new TreeMap<String, String>();
// put your phone numbers

String prefix = ...;
for(Map.Entry<String,String> entry : filterPrefix(nameNum, prefix).entrySet()) {
    System.out.println(entry);
}

如果您希望前缀过滤器不依赖于大小写差异，请为您的地图使用合适的比较器（例如具有合适强度设置的Collator，或String.CASE_INSENSITIVE_ORDER）。

Answer 3

这需要Trie 数据结构。有关 java 实现，请参阅this question。我用this one。

Answer 4

使用guava Multimap 将简化您的解决方案。

键是姓名的第一个字母，值是Collection，包含所有以键（第一个字母）开头的姓名-电话对。

例子：

    public void test(){
      //firstLetter -> list of name-phone pair
      Multimap<String, Pair> mMap =  ArrayListMultimap.create();

      put(mMap, "Brown",  "+1236389023");
      put(mMap, "Bob",    "+1236389023");
      put(mMap, "Harmer", "+1236389023");
      put(mMap, "Harris", "+1236389023");
      put(mMap, "Hawken", "+1236389023");
      put(mMap, "Hosler", "+1236389023");

      //Test
      System.out.println(mMap.get("H"));
   }

   void put(Multimap<String, Pair> mMap, String name, String phone){
      mMap.put(name.substring(0,1), new Pair(name, phone));
   }

   public static class Pair{
      String name;
      String phone;

      public Pair(String name, String phone) {
         this.name = name;
         this.phone = phone;
      }

      @Override
      public String toString() {
         return "Pair [name="+name+", phone="+phone+"]";
      }

}

Answer 5

把它全部放在一个 MultiMap 中（或者只是将一个 List 作为值存储在你的 HashMap 中）。对于“棕色”，存储：

"B"->["Brown"]
"BR"->["Brown"]
"BRO"->["Brown"]

如果您稍后添加“Bradley”：

"B"->["Brown", "Bradley"]
"BR"->["Brown", "Bradley"]
"BRO"->["Brown"]
"BRA"->["Bradley"]

等等……

然后有另一张地图将“布朗”或“布拉德利”映射到电话号码。