K번째 큰 수 찾기

23 Apr 2023 in Algorithm on Java, Coding, Test

문제

설명

현수는 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를 출력합니다.

예시 입력 1

10 3
13 15 34 23 45 65 33 11 26 42

예시 출력 1

143

해결방법

• TreeSet은 기본적으로 중복제거 및 정렬을 지원하는 Collection 이다. (참고)

  // 내림차순 정렬
  TreeSet<Integer> tree = new TreeSet<Integer>(Collections.reverseOrder());
  // 오름차순 정렬
  TreeSet<Integer> tree = new TreeSet<Integer>();
    
  // size, length
  tree.size();
  // 정렬에 맞는 첫번째 값(오름차순이면 제일 작은 값, 내림차순이면 제일 큰 값)
  tree.first();
  // 정렬에 맞는 마지막 값
  tree.last();
  // 요소 제거
  tree.remove(value);
    
  

코드

import java.util.*;

class Main {
    public int solution(int[] arr, int n, int k) {
        int answer = -1;
        // 3장을 뽑아 만들 수 있는 경우의 수를 트리셋에 저장한다.
        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;
    }

    public static void main(String[] args) {
        Main.Main T = new Main.Main();
        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(T.solution(arr, n, k));
    }
}