簡體   English   中英

比較Java中的兩個對象列表

[英]Comparing two list of objects in Java

我有兩個學生對象列表(listA和listB),它們是通過從兩個不同的數據庫中查詢而形成的。 我需要迭代一個列表,並確保它不在另一列表中。

我將以下比較代碼用於同一對象,即覆蓋equals方法並使用for循環進行比較。

假設清單A和清單B可能各有5000行,您能否建議是否有更好的方法來實現這一點?

比較代碼:

for (Student dataA:listA) {
    for (Student dataB:listB) {
        if(dataB.equals(dataA))
            break;                                              
    }
}

學生對象:

public class Student {
    int A;
    int B;
    String C;   

    @Override
    public boolean equals(Object obj) {
        if (obj == this) {
            return true;
        }
        if (obj == null || obj.getClass() != this.getClass()) {
            return false;
        }
        Student student = (Student) obj;
        return A == student.A && B == student.B && C.equals(student.C);
    }
}

編輯說明:ListA和ListB可以具有不同的行數

我建議您使用retainAll方法:

List<Student> listC = new ArrayList();
listC.addAll(listA);
listC.retainAll(listB);   //now listC contains all double students

但是您仍然應該重寫equals方法

您可以使用CollectionUtils (Apache Commons)中的containsAny方法:

if(CollectionUtils.containsAny(listA, listB)){
    break;
}

通用方法是遍歷第一個列表,然后檢查元素是否包含在第二個列表中(如果存在),將元素添加到結果列表中,以下是完整的解決方案

import java.util.ArrayList;
import java.util.List;

public class CompareListofObj {

public static void main(String[] args) {

    List<Student> listStd1 = new ArrayList<Student>();
    List<Student> listStd2 = new ArrayList<Student>();

    Student std1 = new Student(1, 1, "a");
    Student std2 = new Student(2, 1, "b");
    Student std3 = new Student(3, 3, "c");
    Student std4 = new Student(4, 4, "d");
    listStd1.add(std1);
    listStd1.add(std2);
    listStd1.add(std3);
    listStd1.add(std4);

    Student std5 = new Student(1, 1, "a");
    Student std6 = new Student(2, 1, "b");
    Student std7 = new Student(7, 7, "c");
    Student std8 = new Student(8, 8, "d");
    listStd2.add(std5);
    listStd2.add(std6);
    listStd2.add(std7);
    listStd2.add(std8);

    List<Student> listResult = new ArrayList<Student>();

    for (int i = 0; i < listStd1.size(); i++) {
        if (listStd2.contains(listStd1.get(i))) {

            listResult.add(listStd1.get(i));

        } else {

        }
    }

    for (int i = 0; i < listResult.size(); i++) {
        System.out.println("common elt" + listResult.get(i).getA() + ", " + listResult.get(i).getB() + ", "
                + listResult.get(i).getC());
    }

}
}

學生班

package sample;

public class Student {

int A;
int B;
String C;

public Student(int a, int b, String c) {
    super();
    A = a;
    B = b;
    C = c;
}

public int getA() {
    return A;
}

public void setA(int a) {
    A = a;
}

public int getB() {
    return B;
}

public void setB(int b) {
    B = b;
}

public String getC() {
    return C;
}

public void setC(String c) {
    C = c;
}

@Override
public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result + A;
    result = prime * result + B;
    result = prime * result + ((C == null) ? 0 : C.hashCode());
    return result;
}

@Override
public boolean equals(Object obj) {
    if (this == obj)
        return true;
    if (obj == null)
        return false;
    if (getClass() != obj.getClass())
        return false;
    Student other = (Student) obj;
    if (A != other.A)
        return false;
    if (B != other.B)
        return false;
    if (C == null) {
        if (other.C != null)
            return false;
    } else if (!C.equals(other.C))
        return false;
    return true;
}
}

removeAll命令是可以使用的方法,但是列表查找效率不高(線性時間),因此總時間為O(n * m)(n為sizeA,m為sizeB)。 每個條目有5000個條目,可能有點太多。

如果可能的話,您應該將其更改為使用Sets(並在您尚未實現的情況下實現Student類的hashCode和equals方法!):

Set<Student> studentsA = new HashSet<>();
Set<Student> studentsB = new HashSet<>();
studentsA.removeAll(studentsB);

這使您成為O(m * hash(n))。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM