[알고리즘] 04.HashMap, TreeSet / 05. K번째 큰 수

K번째 큰 수

설명

현수는 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 중복을 자동으로 걸러내고, 정렬까지 편하게 할 수 있다

TreeSet.remove(143);
TreeSet.size();
TreeSet.first(); //제일 앞에 있는 수 출력
TreeSet.last();

Tree 구조에 대해서는 아직 좀 헷갈리는 부분이 있는데 좀 더 정리해봐야겠다.

코드

import java.util.Collections;
import java.util.Scanner;
import java.util.TreeSet;
public class Main {

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

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

	public int solution(int n, int k, int[] arr) {
		int answer = -1;

		//중복제거, 정렬
		TreeSet<Integer> tSet = new TreeSet<Integer>(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) {
				answer = x;
			}
		}
		return answer;
	}
}