[알고리즘 문제]K번째 큰 수 - List의 중복 제거, 역순 정렬

▷ 문제

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

▷ 입력 예시

10 3

13 15 34 23 45 65 33 11 26 42

▷ 출력 예시

143

▷ 풀이

import java.util.Scanner;
import java.util.List;
import java.util.ArrayList;
import java.util.stream.Collectors;
import java.util.Comparator;

public class Main {	
    public int solution(int n, int k, int[] arr){
      int answer = -1;
      int cnt = 0;
      List<Integer> list = new ArrayList<Integer>();

      for(int i=0; i<n; i++){
        for(int j=i+1; j<n; j++){
          for(int x=j+1; x<n; x++){
            int sum = arr[i]+arr[j]+arr[x];
            if(!list.contains(sum)) list.add(sum);
          }
        }
      }

      list = list.stream().sorted(Comparator.reverseOrder()).collect(Collectors.toList());
      
      for(int x : list){
	    cnt++;
		  if(cnt == k) return x;
      }
      
      return answer;
    }

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

        kb.close();
        
        System.out.print(main.solution(n, k, arr));
    }
  }

▷ 핵심 포인트

• List에 Stream을 이용하여 중복 제거, 정렬 등 다양한 기능을 사용할 수 있습니다.
• K번째 수가 존재하지 않는다는 점에 유의하여 존재하지 않을 경우 -1을 출력합니다.