Stream 조건부 groupingBy

[주인장]

Stream 조건부 groupingBy

[문제]

학생 리스트를 성별로 분류한 후, 각 그룹에서 점수 기준 내림차순으로 정렬하세요.

[데이터]

new Student(1001, "Kim", "F", 85)

new Student(1002, "Lee", "M", 75)

new Student(1003, "Park", "M", 90)

new Student(1004, "Choi", "F", 88)

[요구사항]

- groupingBy()로 성별 분류

- 각 그룹 내에서 점수 내림차순 정렬

- 결과 출력

[소스]

import java.util.Arrays;

import java.util.Comparator;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

public class ConditionalGroupingBy {

    static class Student {

        int id;

        String name;

        String gender;

        int score;

        public Student(int id, String name, String gender, int score) {

            this.id = id;

            this.name = name;

            this.gender = gender;

            this.score = score;

        }

        public String getName() {

            return name;

        }

        public String getGender() {

            return gender;

        }

        public int getScore() {

            return score;

        }

        @Override

        public String toString() {

            return name + " (" + score + "점)";

        }

    }

    public static void main(String[] args) {

        List<Student> students = Arrays.asList(

            new Student(1001, "Kim", "F", 85),

            new Student(1002, "Lee", "M", 75),

            new Student(1003, "Park", "M", 90),

            new Student(1004, "Choi", "F", 88)

        );

        System.out.println("원본 학생 리스트:");

        students.forEach(System.out::println);

        // groupingBy()로 성별 분류하면서 각 그룹을 점수 내림차순으로 정렬

        Map<String, List<Student>> grouped = students.stream()

                                                      .collect(Collectors.groupingBy(Student::getGender,

                                                                                      Collectors.collectingAndThen(

                                                                                          Collectors.toList(),

                                                                                          list -> list.stream()

                                                                                                    .sorted(Comparator.comparingInt(Student::getScore).reversed())

                                                                                                    .collect(Collectors.toList()))));

        System.out.println("\n성별별 학생 (점수 내림차순):");

        grouped.forEach((gender, studentList) -> {

            System.out.println("[" + (gender.equals("M") ? "남" : "여") + "]");

            studentList.forEach(s -> System.out.println("  " + s));

        });

    }

}