我有两个不同大小的 ArrayList。
ArrayList<String> names = new ArrayList<String>();
ArrayList<User> users = new ArrayList<User>();
用户是具有名称和地址属性的对象。我想获取存储在 ArrayList users 中的 User 对象的名称与存储在 ArrayList 名称中的名称相同的地址。
这可能吗?如果是这样,怎么做?
for (User u : users) {
if (names.contains(u.getName()) {
System.out.println("Name: " + u.getName() + ", address: " + u.getAddress());
}
}
您可以使用 Java 8。
请试试:
users.stream().filter(u -> names.contains(u.getName())).map(
u -> "Name: " + u.getName() + ", address: " + u. getAddress()).forEach(System.out::println);
您需要遍历用户列表并检查每个元素是否包含在名称中。如果其中有很多元素,您可能希望使用 Set 而不是名称列表。
如果您确实在names.contains(..)
内部循环用户,那么您正在执行一个 ^2 循环(大致而言)。
我会将用户地址放入地图中,按名称键入,然后遍历名称列表,将地址从地图中拉出。相反,成本为 2n(大约)。
在 java-8 中,您可以使用如下流和收集器,
逻辑为
1. 在两个列表中查找命令属性
2. 将较小的列表转换为Map<Common_attribute, Actual_List_Object>
O(n) 复杂度
3. 迭代较大的列表并检查地图是否包含,并将其添加到另一个列表中,O(n) 复杂度(地图查找为 ~ O(1)),空间复杂度为 O(n),
否则为 O(n^2)(已接受答案的复杂度)
假设用户的类为
class User {
private String name;
private String getName() {
return this.name;
}
@Override
public String toString() {
return "User{" +
"name='" + name + '\'' +
'}';
}
}
// 达到目的的代码
import java.util.stream.Collectors;
ArrayList<String> names = new ArrayList<>();
ArrayList<User> users = new ArrayList<>();
//adding to name list
names.add("a");
//adding to User list
User user = new User();
user.name = "a";
User user2 = new User();
user2.name = "b";
users.add(user);
users.add(user2);
// Using user's name as key, you can specify any data type as key and create a map
Map<String, User> nameUserMap = users.stream().collect(Collectors.toMap(user1 -> user1.getName(), user1 -> user1));
System.out.println(nameUserMap);
List<User> filteredList = names
.stream()
.map(nameUserMap::get)
.filter(Objects::nonNull)
.collect(Collectors.toList());
System.out.println(filteredList);
输出:
{a=User{name='a'}, b=User{name='b'}}
[User{name='a'}]