Java ArrayList 到 HashMap 的映射

Question

我有页面被给予 ArrayList;其中每个文档都有一个称为类型的属性。

我不知道独特类型或文档的数量。

我想将此 ArrayList 排序为 HashMap但我在理解它时遇到了一些困难。

一些伪代码希望

for (int i = 0; i < documents.size(); i++) 
{
   if there is an array for documents[i].type
   add to this array
   else create a new array for this type
   add document[i].type and the array of documents with matching type to the hashmap
}

我知道这是错误的方法并且显然行不通。我愿意接受任何建议。

谢谢

Answer 1

// create the map to store stuff, note I'm using a List instead of an array
// in my opinion it's a bit cleaner
Map<String, List<Document>> map = new HashMap<String, List<Document>>();

// now iterate through each document
for(Document d : documents){

    // check to see if this type is already known
    List<Document> list = map.get(d.type);

    if(list == null){
        // list is null when it wasn't found in the map
        // this is a new type, create a new list
        list = new ArrayList<Document>();

        // store the list in the map
        map.put(d.type, list);
    }

    // finally, whether we got a hit or a miss, we want
    // to add this document to the list for this type
    list.add(d);
}

Answer 2

我认为您要查找的术语不是按类型排序，而是按类型索引。 Guava 的 Multimap 接口旨在将键映射到多个值，而无需处理集合的所有麻烦价值观。特别是，Guava 有一个方法，旨在完全执行您想要执行的操作：

List<Document> documents = ...
ImmutableListMultimap<Type, Document> typeIndex = Multimaps.index(documents,
    new Function<Document, Type>() {
      public Type apply(Document input) {
        return input.getType();
      }
    });

for(Type type : typeIndex.keySet()) {
  ImmutableList<Document> documentsWithType = typeIndex.get(type);
  ...
}

这与执行： 几乎相同，

ListMultimap<Type, Document> typeIndex = ArrayListMultimap.create();
for(Document document : documents) {
  typeIndex.put(document.getType(), document);
}

只是生成的多重映射是不可变的。另请注意，上面的内容几乎与马克的示例完全相同。