[PS] 인프런 강의 - HashMap, HashSet, TreeSet 5. 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에 카드의 합을 저장한다.

 

주의!!! 헷갈리지 말것 ! 뽑는 카드의 수는 3장으로 정해져있으므로 3중 for문으로 푼다.

 

k번째로 큰 수를 출력해야 하므로 내림차순으로 정렬한다.

 

코드

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

public class Main {
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		
		int n = in.nextInt();
		int k = in.nextInt();
		int[] card = new int[n];
		TreeSet<Integer> Tset = new TreeSet<>(Collections.reverseOrder());	
	
		for(int i=0; i<n; i++) {
			card[i] = in.nextInt();
			
		}
		//해쉬맵에서 k개를 뽑는 모든 경우의 수
		int answer = -1;
		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(card[i] + card[j] + card[l]);
				}
			}
		}

		int cnt=0;
		for(int x : Tset) {
			cnt++;
			if(cnt == k) answer = x;
		}
		
		System.out.print(answer);
		
		
	}
}