在java中的列表中搜索对象的最快方法

Question

我有一个结构：

public class DataItem {
    public int wordID, categoryID, documentID, count;
}

我有一个如下列表：

final public ArrayList<DataItem> data = new ArrayList<>();

我已经写了一个在里面搜索的方法：

public DataItem FindDataItem(final int wordID, final int categoryID, final int documentID)
{
    for(DataItem dataItem : data)
        if(dataItem.wordID == wordID && dataItem.documentID == documentID && dataItem.categoryID == categoryID)
            return dataItem;
    return null;
}

但是太慢了。如何加快速度？

我正在考虑四个 HashMap 相互内部，但我想像数据库表一样使用这些数据，因此很难在 HashMap 中按计数分组

我也在考虑ParalellStream，但是不知道怎么用。看起来很复杂。但它仍然是 O(n)。

我也在考虑使用数据库。但我不想有 IO。我希望它全部在 RAM 中。

请指导我完成这个。

Answer 1

正如@ShreyasSarvothama 在 cmets 中所说，检索值的最快方法是使用 Map。

我认为您可以使用一个映射，其键是使用您用作 find 方法参数的值计算的（考虑到它们的组合给出了 DataItem 的唯一标识符）。

import java.util.*;
import java.util.stream.*;

public class Test {

    private class DataItem {
        public int wordID, categoryID, documentID, count;

        public DataItem(int w, int c, int d) {
            wordID = w;
            categoryID = c;
            documentID = d;
        }

        public String toString() {
            return "wordID:" + wordID + " categoryID:" + categoryID + " documentID:" + documentID;
        }
    }

    private Map<Integer, DataItem> map;

    public void setList(List<DataItem> list) {
        this.map = list.stream().collect(Collectors.toMap(dataItem -> dataItem.wordID * dataItem.categoryID * dataItem.documentID, dataItem -> dataItem));        
    }

    public DataItem getDataItem(int wordID, int categoryID, int documentID) {
        return map.get(wordID * categoryID * documentID);
    }

    public static void main(String[] args) {
        Test t = new Test();
        t.setList(Arrays.asList(t.new DataItem(1,2,3), t.new DataItem(2,3,4), t.new DataItem(3,3,4)));
        System.out.println(t.getDataItem(2,3,4));
    }
}

希望对你有帮助。

Answer 2

以下是使用并行流的方法：

public DataItem FindDataItem(final int wordID, final int categoryID, final int documentID) {
    return data.parallelStream()
            .filter(dataItem -> dataItem.wordID == wordID
                    && dataItem.documentID == documentID
                    && dataItem.categoryID == categoryID)
            .findAny()
            .orElse(null);
}

正如你所说，它不会影响时间复杂度，但它可以根据可用线程的数量加快操作。