Notice
Recent Posts
Recent Comments
Link
«   2024/09   »
1 2 3 4 5 6 7
8 9 10 11 12 13 14
15 16 17 18 19 20 21
22 23 24 25 26 27 28
29 30
Tags more
Archives
Today
Total
관리 메뉴

ultra_dev

K번째 큰 수 (TreeSet) 본문

알고리즘

K번째 큰 수 (TreeSet)

ultra_dev 2023. 3. 29. 15:56

설명

현수는 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

☑️Collections.reverseOrder()

기본값이 오름차순. 지금은 큰값이 앞에 오는 내림차순 필요하니까.Collections.reverseOrder()

☑️Tset.remove(143);

Tset.size()

Tset.first() →정렬시 제일 앞, 오름차순이면 최소값, 내림차순이면 최대값이겠지~

Tset.Last() →정렬시 제일 뒤

📌3중 for문

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

class Main {

    public int solution(int[] arr, int n, int k) {
        int answer = -1; //k번째 수가 존재하지 않으면 -1 리턴하라고 했으니
        TreeSet<Integer> Tset = new TreeSet<>(
            Collections.reverseOrder()); // 기본값이 오름차순. 지금은 큰값이 앞에 오는 내림차순 필요하니까.Collections.reverseOrder()
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < n;
                j++) {   // i+1 , j+1이런식으로 안겹치게 3중 for문! n은 안건드려도됨 why? 만약 9까지 돌 때 i가 8이면 j가 9 l이 10이돼야하니 거짓이 되버림! add안됨!
                //정 찝찝하면 i쪽 n-2 j쪽 n-1해도 됨 !
                for (int l = j + 1; l < n; l++) {
                    Tset.add(arr[i] + arr[j] + arr[l]);
                }
            }

        }
        int cnt = 0;
        for (int x : Tset) {
            cnt++; //순위라고 생각하면 됨. 맨위에놈은 1로 되겠지 cnt++;이 앞에 있으니
            if (cnt == k) {
                return x;
            }
        }
        return answer;
    }

    public static void main(String[] args) {
        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.println(T.solution(arr, n, k));
    }
}

'알고리즘' 카테고리의 다른 글

괄호문자제거(Stack)  (0) 2023.03.30
올바른 괄호(Stack)  (0) 2023.03.30
매출액의 종류  (0) 2023.03.29
아나그램(해쉬)  (0) 2023.03.27
학급회장(HashMap)  (0) 2023.03.27
Comments