[Inflearn] 자바 알고리즘 문제풀이 #04-05 5. k번째 큰 수 (TreeSet)

문제

설명

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

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

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

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

 

입력 

첫 줄에 자연수 N(3<=N<=100)과 K(1<=K<=50) 입력되고, 그 다음 줄에 N개의 카드값이 입력된다.

 

출력 

첫 줄에 K번째 수를 출력합니다. K번째 수가 존재하지 않으면 -1를 출력합니다.

 

 

 

내 풀이

import java.util.*;

public class Main4_5 {
    public int Solution(int n, int k, int[] arr) {
        ArrayList<Integer> al = new ArrayList<>();
        ArrayList<Integer> al_unique = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            for (int j = i+1; j < n; j++) {
                for (int l = j+1; l < n; l++) {
                    al.add(arr[i]+arr[j]+arr[l]);
                }
            }
        }
        // 중복되지 않는 수만 al_unique에 저장
        for(int x : al) {
            if(!al_unique.contains(x))
                al_unique.add(x);
        }
        // 내림차순 정렬
        al_unique.sort(Comparator.reverseOrder()); // 혹은 Collections.sort(al, Collections.reverseOrder());
        int cnt=0;
        for(int x : al_unique) {
            cnt++;
            if(cnt==k) return x;
        }
        return -1;
    }

    public static void main(String[] args) {
        Main4_5 T = new Main4_5();
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        System.out.println(T.Solution(n, k, arr));
    }
}

몇 번을 고친 끝에 정답~

TreeSet에 대해 몰랐기 때문에

.contains(x)을 이용해 중복을 제거해주었고, Comparator.reverseOrder()로 ArrayList를 내림차순 정렬해주었다.

 

 

 

선생님 풀이

import java.util.*;

public class Main4_5 {
    public int Solution(int n, int k, int[] arr) {
        int answer=-1;
        TreeSet<Integer> Tset = new TreeSet<>(Collections.reverseOrder());
        for (int i = 0; i < n; i++) { // 끝까지 돌아도 상관없음. 자동으로 끝남
            for (int j = i+1; j < n; j++) {
                for (int l = j+1; l < n; l++) {
                    Tset.add(arr[i]+arr[j]+arr[l]);
                }
            }
        }
        int cnt=0;
        for(int x : Tset) {
            cnt++;
            if(cnt==k) return x;
        }
        return answer; // for문에서 끝까지 거짓일시 -1 반환
    }

    public static void main(String[] args) {
        Main4_5 T = new Main4_5();
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        System.out.println(T.Solution(n, k, arr));
    }
}

여기서 알아두어야 할 것

Set : 중복을 허용하지 않는 자료구조

TreeSet : 내림차순이나 오름차순으로 중복없이 정렬하는 자료구조 (중복제거+정렬)

 

TreeSet<Integer> Tset = new TreeSet<>(Collections.reverseOrder());

이런 단순한 방법이 있다니ㅎㅎㅎㅎ 코테도 자바 언어에 대해 아는만큼 보이고, 풀 수 있는 것 같다.

 

 

 

결과