java流列表,Java流 - 通过嵌套列表分组（按第二顺序列出）

I have the following data structure -

List of Students that each holds a lists of States that each holds a list of cities.

public class Student {

private int id;

private String name;

private List states = new ArrayList<>();

}

public class State {

private int id;

private String name;

private List Cities = new ArrayList<>();

}

public class City {

private int id;

private String name;

}

I want to get the following.

Map citiesIdsToStudensList;

I write the following

Map> statesToStudentsMap = students.stream()

.flatMap(student -> student.getStates().stream())

.flatMap(state -> state.getCities().stream())

.collect(Collectors.groupingBy(City::getId, Collectors.mapping(x -> x.getId(), Collectors.toList())));

But it doesn't get me the result I want.

解决方案

Using the Stream API, you'll need to flat map twice and map each intermediate student and city into a tuple that is capable of holding the student.

Map> citiesIdsToStudentsList =

students.stream()

.flatMap(student -> student.getStates().stream().map(state -> new AbstractMap.SimpleEntry<>(student, state)))

.flatMap(entry -> entry.getValue().getCities().stream().map(city -> new AbstractMap.SimpleEntry<>(entry.getKey(), city)))

.collect(Collectors.groupingBy(

entry -> entry.getValue().getId(),

Collectors.mapping(Map.Entry::getKey, Collectors.toList())

));

However, it would maybe be cleaner to use nested for loops here:

Map> citiesIdsToStudentsList = new HashMap<>();

for (Student student : students) {

for (State state : student.getStates()) {

for (City city : state.getCities()) {

citiesIdsToStudentsList.computeIfAbsent(city.getId(), k -> new ArrayList<>()).add(student);

}

}

}

This leverages computeIfAbsent to populate the map and creates a list of each student with the same city id.