[JAVA] 알고리즘 문제풀이 입문 4-5

강의: https://inf.run/w779

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

 

4-5번

import java.util.Collections;

import java.util.Scanner;

import java.util.TreeSet;

public class Main {

  public int solution(int n, int k, int[] arr){

    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 -1;

  }

  public static void main(String[] args){

    Main t = new Main();

    Scanner in = new Scanner(System.in);

    int n = in.nextInt();

    int k = in.nextInt();

    int[] arr = new int[n];

    for(int i = 0; i < n; i++){

      arr[i] = in.nextInt();

    }

    in.close();

    System.out.print(t.solution(n, k, arr));

  }

}