java List包装类型，封装对象问题

Question

比如，有两个类A、B： public class A{ private int a1; private int a2; private B b; //getter setter } public class B{ private int b1; private int b2; private int a1; //getter setter } public static void main(String[] args) { Set<A> arr = service1.findList(); List<B> arr2 = service2.findList(); } 现在我需要，arr中item的a1如果等于arr2的item2的a1，就把item2放到item的b属性中，正常来说这样子 public static void main(String[] args){ Set<A> arr = service1.findList(); List<B> arr2 = service2.findList(); Iterator<A> it = arr .iterator(); while(it.hasNext()){ A a = it.next(); arr2.forEach(b->{ if(a.getA1() == b.getA1()){ a.setB(b); return; } }); } } 但是一坨这样子双循环看到都恶心，有没有好的方法，目前用jdk8

Answer 1

单纯从效率上考虑，你这样子双循环的复杂度是O(n^2)，其实可以优化的：

先做一个Map<Integer, B>，即a1->B的映射，再遍历Set<A>就可以了。

Answer 2

一楼正解，用java8的lambda表达式可以很简练的实现：

Map<Integer, B> bMap = arr2.stream().collect(Collectors.toMap(B::getA1, Function.identity()));
arr.forEach(a -> {
            B b = bMap.get(a.getA1());
            a.setB(b);
        });