설명

현수는 1부터 100사이의 자연수가 적힌 N장의 카드를 가지고 있습니다. 같은 숫자의 카드가 여러장 있을 수 있습니다.

현수는 이 중 3장을 뽑아 각 카드에 적힌 수를 합한 값을 기록하려고 합니다. 3장을 뽑을 수 있는 모든 경우를 기록합니다.

기록한 값 중 K번째로 큰 수를 출력하는 프로그램을 작성하세요.

만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰 값은 22입니다.

코드

public class KNumber {
    public static void main(String[] args){
        Scanner in=new Scanner(System.in);
        int input1 = in.nextInt();
        int[] arr = new int[input1];
        int input2 = in.nextInt();
        for(int i=0; i<input1; i++){
            arr[i] = in.nextInt();
        }
        int solution = solution(input2, arr);
        System.out.println(solution);
    }

    public static int solution(int k, int[] arr){
        Set<Integer> set = new TreeSet<>(Comparator.reverseOrder());
        int length = arr.length;

        for(int i=0; i<length; i++){
            for(int j=i+1; j<length; j++){
                for(int h=j+1; h<length; h++){
                    set.add(arr[i]+arr[j]+arr[h]);
                }
            }
        }

        for(int i : set){
            if(k==1) return i;
            else k--;
        }

        return -1;
    }
}

프로그래머스의 K번째의 수 문제와 혼동하여 문제를 이해하지 못해 시간이 오래 걸렸다... 설명도 내가 이해력이 딸리는건지 잘 이해가 안되더라 ㅜㅜ 그래서 결국 강의를 보니 배열의 3개의 수를 합한 모든 경우의 수 중에서 3번째 큰 수를 출력하는 문제였다... sliding window나 two poniter 알고리즘으로만 문제를 풀려고 생각했더니 절대 안풀리는게 당연하지... 그래서 또 결국 강의를 보니 3중 for문으로 모든 경우의 수만큼 반복 시킨다.

그리고 여기서 for문의 길이에 length 값 대신 i의 경우 length-2, j는 length-1만큼 반복키게 코드를 짜도 상관 없지만 위와 같이 코드를 작성해도 큰 문제는 되지 않는다. 왜냐하면 이미 i가 length-2일때 h는 length 값을 가지고 for문이 실행되지 않으므로 반복문에서는 크게 차이가 나지 않는다. 하지만 진짜 정말 마지막 최적화까지 생각한다면 고려해볼 만 한 문제다.

해당 문제는 Set에 대한 이해도를 높이기 위한 문제였음!