使用列表获取匹配项索引的有效方法

Question

我有两个列表 A 和 B。我想找出 A 中与 listB 的元素匹配的元素的索引。像这样的：

ArrayList listA = new ArrayList();
listA.add(1);listA.add(2);listA.add(3);listA.add(4);
ArrayList listB = new ArrayList();
listB.add(2);listB.add(4);
ArrayList listC = new ArrayList();
for(int i=0; i<listB.size();i++) {
   int element = listB.get(i);
   for(int j=0; j<listA.size(); j++) {
      if(listA.get(j) == element) listC.add(j);
   }
}

我想这是一种丑陋的做法。查找与 B 中的所有元素匹配的 A 的所有索引的最佳方法是什么？我相信 collections api 中存在一个名为 containsAll 的方法 - 不要认为它返回匹配的索引。

Answer 1

如果您必须使用ArrayList，您可以从ArrayList 创建一个HashSet。这将调用contains O(1)。创建HastSet 需要 O(n)。如果您可以以HashSet 开头，那就最好了。

public static void main(String[] args) 
{
    List listA = new ArrayList();
    listA.add(1);
    listA.add(2);
    listA.add(3);
    listA.add(4);
    List listB = new ArrayList();
    listB.add(2);
    listB.add(4);
    Set hashset = new HashSet(listA);

    for(int i = 0; i < listB.size(); i++) 
    {
        if(hashset.contains(listB.get(i))) 
        {
            listC.add(i);
            System.out.println(i);
        }
    }   
}

Answer 2

Guava 库带有一个方法

"SetView com.google.common.collect.Sets.intersection(Set a, Set b)

这将给出两个集合中包含的元素，但不给出索引。虽然之后应该很容易获取索引。

Answer 3

简单：

List<Integer> listA = new ArrayList<Integer>();
listA.add(1);
listA.add(2);
listA.add(3);
listA.add(4);

List<Integer> listB = new ArrayList<Integer>();
listB.add(2);
listB.add(4);

List<Integer> listC = new ArrayList<Integer>();

for ( Integer item : listA ) {
    int index = listB.indexOf( item );
    if ( index >= 0 ) {
        listC.add(index);
    }
}

但这只有在没有重复的情况下才有效，如果有重复的索引，你必须按照你的方式来做，浏览完整的列表。

编辑

我以为你想要的是元素，而不是索引，集合不会给你索引，只有元素。

Answer 4

假设没有重复值，为什么不使用ArrayList.indexOf？

---

public final class ArrayListDemo {
    public static void main(String[]args){
        findIndices(createListA(), createListB());      
    }

    private static final List<Integer> createListA(){
        List<Integer> list = new ArrayList<Integer>();
        list.add(1);
        list.add(3);
        list.add(5);

        return list;
    }

    private static final List<Integer> createListB(){
        List<Integer> list = new ArrayList<Integer>();
        list.add(0);
        list.add(2);
        list.add(3);
        list.add(4);

        return list;
    }

    private static void findIndices(List<Integer> listA, List<Integer> listB){
        for(int i = 0; i < listA.size(); i++){  
            // Get index of object in list b
            int index = listB.indexOf(listA.get(i));

            // Check for match
            if(index != -1){
                System.out.println("Found match:");
                System.out.println("List A index = " + i);
                System.out.println("List B index = " + index);
            }
        }
    }
}

输出

Found match:
List A index = 1
List B index = 2

Answer 5

如果列表 A 和列表 B 以相同的顺序排序（我假设升序，但降序也可以）这个问题有一个 O(n) 的解决方案。下面是一些（非正式的、未经测试的）代码。当循环退出时，indexMap 应该包含列表 A 中与列表 B 中的元素匹配的每个元素的索引以及列表 B 中匹配元素的索引。

国际电流A； 国际电流B； int listAIndex = 0; int listBIndex = 0; Map indexMap = new HashMap(); currentA = listA.get(listAIndex); currentB = listB.get(listBIndex); while ((listAIndex currentB) { ++列表B索引； } }

Answer 6

使用Apache CollectionUtils，有很多选择