[BackJoon]25305번 스트림으로 풀기

문제

 

2022 연세대학교 미래캠퍼스 슬기로운 코딩생활에 N명의 학생들이 응시했다.
이들 중 점수가 가장 높은 k명은 상을 받을 것이다. 이 때, 상을 받는 커트라인이 몇 점인지 구하라.
커트라인이란 상을 받는 사람들 중 점수가 가장 가장 낮은 사람의 점수를 말한다.

 

입력

 

첫째 줄에는 응시자의 수 N과 상을 받는 사람의 수 k가 공백을 사이에 두고 주어진다.
둘째 줄에는 각 학생의 점수 x가 공백을 사이에 두고 주어진다.

 

출력

 

상을 받는 커트라인을 출력하라.

 

풀이

 

상을 받는 사람 중 가장 낮은 사람의 점수를 찾는 문제다.

 

예를 들어 5명 중 2명이 상을 받게 된다면 정렬 수 두 번째로 큰 점수를 찾으면 된다.

 

스트림을 사용해서 정렬 후, 상을 받을 사람의 수 k 에서 1을 뺀 만큼 스킵한 뒤 커트라인을 찾았다.

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Scanner;

public class Prob25305Stream {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();
        int k = sc.nextInt();
        List<Integer> list = new ArrayList<>();

        for (int i = 0; i < n; i++) {
            list.add(sc.nextInt());
        }

        int result = list.stream()
                .sorted(Comparator.reverseOrder())
                .mapToInt(Integer::intValue)
                .skip(k - 1)
                .findFirst()
                .getAsInt();

        System.out.println(result);
    }
}