5 mjj111

[mjj111]

🔗 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/258709
주사위 고르기

✔️ 소요된 시간

1시간 20분

✨ 코드 설명

해당 문제를 거시적으로 설명하면,
주사위 배열을 입력받고 초기화를 수행한 후,
choiceDice 메서드를 호출하여 최적의 그룹을 찾습니다.
최종 결과를 answer 배열에 저장합니다.

여기서 choiceDice 메서드를 통해 어떻게 최적의 그룹을 찾는가 순차적으로 설명드리겠습니다.

choiceDice 메서드

재귀적으로 주사위를 선택하여 두 그룹으로 나누는 모든 조합을 생성합니다.
각 조합에 대해 calculateWinningPercent 메서드를 호출하여 승리 확률을 계산하고, 최대 승리 확률을 갱신합니다.

    public static void choiceDice(int depth, int s) {
        if (depth == N / 2) {
            int winning = calculateWinningPercent();
            if (max < winning) {
                max = winning;
                for (int i = 0; i < choice.size(); i++) {
                    answer[i] = choice.get(i) + 1;
                }
            }
            return;
        }
        
        for (int i = s; i < N; i++) {
            choice.add(i);
            choiceDice(depth + 1, i + 1);
            choice.remove(choice.size() - 1);
        }
    }

calculateWinningPercent 메서드

현재 선택된 조합에 대해 두 그룹(arrA, arrB)의 숫자 합을 계산합니다.
그룹 B를 정렬한 후, 그룹 A의 각 숫자에 대해 그룹 B에서 이분 탐색으로 작은 숫자의 개수를 찾아 승리 횟수를 계산합니다.

    private static int calculateWinningPercent() {
        int count = 0;

        makeArrAB(); 

        Collections.sort(arrB);
        for (int i = 0; i < arrA.size(); i++) {
            // 이분 탐색으로 arrB에 number 보다 작은 숫자가 몇개 있는지 구한다.
            int number = arrA.get(i);

            int left = 0, right = arrB.size() - 1;

            int index = Integer.MIN_VALUE;
            while (left <= right) {
                int middle = (left + right) / 2;

                if (arrB.get(middle) < number) { // 조건 만족: number 보다 작아야 함!! 같아도 안됨.
                    left = middle + 1;
                    index = Math.max(index, middle);
                } else {
                    right = middle - 1;
                }
            }
            if (index != Integer.MIN_VALUE) {
                count += index + 1;
            }
        }
        return count;
    }

makeArrAB 메서드

현재 선택된 주사위를 기준으로 두 그룹(arrA, arrB)을 만듭니다.
각각의 그룹에 대해 가능한 모든 숫자 합을 생성합니다.

    private static int calculateWinningPercent() {
        int count = 0;

        makeArrAB(); 

        Collections.sort(arrB);
        for (int i = 0; i < arrA.size(); i++) {
            // 이분 탐색으로 arrB에 number 보다 작은 숫자가 몇개 있는지 구한다.
            int number = arrA.get(i);

            int left = 0, right = arrB.size() - 1;

            int index = Integer.MIN_VALUE;
            while (left <= right) {
                int middle = (left + right) / 2;

                if (arrB.get(middle) < number) { // 조건 만족: number 보다 작아야 함!! 같아도 안됨.
                    left = middle + 1;
                    index = Math.max(index, middle);
                } else {
                    right = middle - 1;
                }
            }
            if (index != Integer.MIN_VALUE) {
                count += index + 1;
            }
        }
        return count;
    }

makeArr 메서드

주어진 주사위 배열에서 가능한 모든 숫자 합을 재귀적으로 계산하여 리스트에 저장합니다.

    public static void makeArr(int depth, int[][] dice, int sum, List<Integer> arr) {
        if (depth == N / 2) {
            arr.add(sum);
            return;
        }
        for (int i = 0; i < 6; i++) {
            int newSum = sum + dice[depth][i];
            makeArr(depth + 1, dice, newSum, arr);
        }
    }

📚 새롭게 알게된 내용 & 하소연

요즘 알고리즘 기출 문제에서 이분탐색이 나오면 항상 조합 관련된 복합 문제가 자주 나오네요.
추세가 여러 알고리즘을 함께 사용할 수 있는 능력을 요구하나 봅니다.
이런 문제를 좀 찾아다녀야겠습니다 🤔
+
순차적으로 메서드를 생성해가면서 구현했는데, 중간중간 로직이 잘 구현이 안되네요 ㅠ
요즘 개발만해서 그런지 항상 말랑말랑한 상태를 유지할 수 있도록
이런 구현 문제 위주로 계속 풀어봐야겠습니다

[9kyo-hwang]

헣허 예전에 제가 풀었던 문제... 정말 머리 싸매가면서 풀었었는데요...
눈물겨운 저의 똥꼬쇼를 봐주시면 감사하겠습니다...

[xxubin04]

대략 2시간 좀 넘게 고민한 것 같은데 코드가 많이 지저분하네요 ...
테케도 8개 정도 시간초과났지만 우선 올리도록 하겠습니다!

from itertools import combinations, product

def solution(dice):
    n = len(dice)
    list_comb = list(combinations(dice, n//2))
    dice_num = list(combinations([p for p in range(1, n+1)], n//2))
    result = [0] * len(list_comb)

    for a in range(len(list_comb)):
        b = [x for x in dice if not x in list_comb[a]]
        A_comb = list(product(*list_comb[a]))
        B_comb = list(product(*b))
        for i in range(len(A_comb)):
            for j in range(len(B_comb)):
                if sum(A_comb[i]) > sum(B_comb[j]):
                    result[a] += 1

    answer = dice_num[result.index(max(result))]
    return answer

뭔가 더 깔끔하게 풀 수 있을 것 같은데 아직은 역부족이네요. 더 공부해야겠습니다!!
수고 많으셨습니다~!