[Algorithm/Java] 알고리즘 자바 K번째 큰 수 (코딩테스트, TreeSet, 트리, 셋, 합의 순위, rank)

---

1. 문제 설명

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

 

 

2. 문제 코드 및 풀이 설명

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

public class Ranking {
  public static void main(String[] args) {
    Ranking T = new Ranking();
    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));
  }

  // 해결방법
  // 모든 합의 경우를 구하기 위해 TreeSet에 대입하여 중복을 제거한다
  // 내림차순으로 정렬하여 k번째 값을 출력한다
  public int solution(int[] arr, int n, int k) {
    int answer = -1;
    // 중복제거를 위한 TreeSet
    TreeSet<Integer> Tset = new TreeSet<>(Collections.reverseOrder());
    // TreeSet에 모두 대입
    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]);
        }
      }
    }
    // TreeSet의 k번째 값 출력 = k번째로 큰 값
    for (int x : Tset) {
      k--;
      if(k == 0) {
        answer = x;
        break;
      }
    }
    return answer;
  }
}

 

3. 출력 예시

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

출력
143

 

 

---

위의 내용은 인프런에서 수강할 수 있는 김태원님의 자바 알고리즘 문제풀이 강의를 바탕으로 공부한 내용을 정리한 내용입니다!

https://inf.run/iAi6

자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의