4-5. K번째로 큰 수

package HashMap그리고TreeSet4장.K번째로큰수4다시5.my;

import java.util.*;
import java.util.stream.Collectors;

public class Main {

    /**
     * Set 자료구조를 사용하면 간단하게 풀리는 문제
     * Set 은 중복을 허용하지 않는다. 거기다 오름차순 내림차순으로 정렬하고싶다면 TreeSet 을 사용하면 된다.
     * TreeSet 은 선언하고 입력받는 것들을 정렬해준다.
     * 비슷한 것으로 TreeMap 이 있다.
     *
     * 자세한 내용은 선생님의 풀이 참고고
    */

    public static void main(String[] args) {
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        int k = kb.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = kb.nextInt();
        }

        System.out.println(solution(n, k, arr));
    }

    private static int solution(int n, int k, int[] arr) {
       int answer = 0;
       HashSet<Integer> set = new HashSet<>();

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                for (int l = 0; l < n; l++) {
                    int sum = 0;
                    if (i != j && j != l && i != l) {
                        sum += arr[i] + arr[j] + arr[l];
                        set.add(sum);
                    }
                }
            }
        }

        for (Integer integer : set) {
            System.out.print(integer + " ");
        }

        return answer;
    }
}

 

package HashMap그리고TreeSet4장.K번째로큰수4다시5.teacher;

import java.util.Collections;
import java.util.Scanner;
import java.util.TreeSet;

public class Main {

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

    private static int solution(int[] arr, int n, int k) {
        int answer = -1;
        TreeSet<Integer> treeSet = new TreeSet<>(Collections.reverseOrder());
        // 기본적으로 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++) {
                    treeSet.add(arr[i] + arr[j] + arr[l]);
                }
            }
        }

        int cnt = 0;
        treeSet.remove(143);    // 해당 값 지우기
        System.out.println(treeSet.size());     // 키가 몇개 있는지
        treeSet.first();    // 가장 첫번째 값
        treeSet.last();     // 가장 마지막 값
        for (int x : treeSet) {
            System.out.println(x);
            cnt++;
            if (cnt == k) {
                return x;
            }
        }

        return answer;
    }
}