ABOUT ME

-

Today
Yesterday
Total
  • [코테] java - K번째 큰수 ( TreeSet )
    알고리즘/코테 2024. 11. 8. 15:30

     

    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

     

     

     

    문제해결 전략:

    1. TreeSet을 활용해서 중복 값제거 + 정렬

    2. 3중 for문 돌 때 i<n-2, j<n-1 로 설정할 필요없음.

    -> i가 n-1, n이될 때는 j=n, k=n+1과 같이되므로 2번째, 3번째 for문이 유효하지않게 되면서  3번째 for문 내부의 Tset 명령문이 실행되지 않기때문에 틀리지 않은 코드가 됨.

     

    3. forEach문 이용하여 Tset을 순회하며 3번째 값을 찾음.

    Scanner sc = new Scanner(System.in);
    
    int n = sc.nextInt();
    int m = sc.nextInt();
    
    int[] a = new int[n];
    for(int i=0; i<n; i++){
        a[i] = sc.nextInt();
    }
    
    // TreeSet활용(중복제거,내림차순 정렬)
    TreeSet<Integer> Tset = new TreeSet<>(Collections.reverseOrder());
    
    for(int i=0; i<n; i++){        // i<n-2로 안해도됨.
        for(int j=i+1; j<n; j++){  // j<n-1로 안해도됨.
            for(int k=j+1; k<n; k++){
                Tset.add(a[i]+a[j]+a[k]);
            }
        }
    }
    
    int answer= -1;
    int count =0;
    
    for(int x: Tset){
        count++;
        if(count==m){
            answer = x;
        }
    }
    System.out.println(answer);
Designed by Tistory.