본문 바로가기
개발/알고리즘

[프로그래머스][LEVEL2] 양궁대회

by ISA(류) 2022. 1. 22.

# 문제 원문

https://programmers.co.kr/learn/courses/30/lessons/92342

 

코딩테스트 연습 - 양궁대회

문제 설명 카카오배 양궁대회가 열렸습니다. 라이언은 저번 카카오배 양궁대회 우승자이고 이번 대회에도 결승전까지 올라왔습니다. 결승전 상대는 어피치입니다. 카카오배 양궁대회 운영위원

programmers.co.kr

카카오배 양궁대회가 열렸습니다.
라이언은 저번 카카오배 양궁대회 우승자이고 이번 대회에도 결승전까지 올라왔습니다. 결승전 상대는 어피치입니다.
카카오배 양궁대회 운영위원회는 한 선수의 연속 우승보다는 다양한 선수들이 양궁대회에서 우승하기를 원합니다. 따라서, 양궁대회 운영위원회는 결승전 규칙을 전 대회 우승자인 라이언에게 불리하게 다음과 같이 정했습니다.

  1. 어피치가 화살 n발을 다 쏜 후에 라이언이 화살 n발을 쏩니다.
  2. 점수를 계산합니다.
    1. 과녁판은 아래 사진처럼 생겼으며 가장 작은 원의 과녁 점수는 10점이고 가장 큰 원의 바깥쪽은 과녁 점수가 0점입니다. 
    2. 만약, k(k는 1~10사이의 자연수)점을 어피치가 a발을 맞혔고 라이언이 b발을 맞혔을 경우 더 많은 화살을 k점에 맞힌 선수가 k 점을 가져갑니다. 단, a = b일 경우는 어피치가 k점을 가져갑니다. k점을 여러 발 맞혀도 k점 보다 많은 점수를 가져가는 게 아니고 k점만 가져가는 것을 유의하세요. 또한 a = b = 0 인 경우, 즉, 라이언과 어피치 모두 k점에 단 하나의 화살도 맞히지 못한 경우는 어느 누구도 k점을 가져가지 않습니다.
      • 예를 들어, 어피치가 10점을 2발 맞혔고 라이언도 10점을 2발 맞혔을 경우 어피치가 10점을 가져갑니다.
      • 다른 예로, 어피치가 10점을 0발 맞혔고 라이언이 10점을 2발 맞혔을 경우 라이언이 10점을 가져갑니다.
    3. 모든 과녁 점수에 대하여 각 선수의 최종 점수를 계산합니다.
  3. 최종 점수가 더 높은 선수를 우승자로 결정합니다. 단, 최종 점수가 같을 경우 어피치를 우승자로 결정합니다.

현재 상황은 어피치가 화살 n발을 다 쏜 후이고 라이언이 화살을 쏠 차례입니다.
라이언은 어피치를 가장 큰 점수 차이로 이기기 위해서 n발의 화살을 어떤 과녁 점수에 맞혀야 하는지를 구하려고 합니다.

화살의 개수를 담은 자연수 n, 어피치가 맞힌 과녁 점수의 개수를 10점부터 0점까지 순서대로 담은 정수 배열 info가 매개변수로 주어집니다. 이때, 라이언이 가장 큰 점수 차이로 우승하기 위해 n발의 화살을 어떤 과녁 점수에 맞혀야 하는지를 10점부터 0점까지 순서대로 정수 배열에 담아 return 하도록 solution 함수를 완성해 주세요. 만약, 라이언이 우승할 수 없는 경우(무조건 지거나 비기는 경우)는 [-1]을 return 해주세요.


제한사항

  • 1 ≤ n ≤ 10
  • info의 길이 = 11
    • 0 ≤ info의 원소 ≤ n
    • info의 원소 총합 = n
    • info의 i번째 원소는 과녁의 10 - i 점을 맞힌 화살 개수입니다. ( i는 0~10 사이의 정수입니다.)
  • 라이언이 우승할 방법이 있는 경우, return 할 정수 배열의 길이는 11입니다.
    • 0 ≤ return할 정수 배열의 원소 ≤ n
    • return할 정수 배열의 원소 총합 = n (꼭 n발을 다 쏴야 합니다.)
    • return할 정수 배열의 i번째 원소는 과녁의 10 - i 점을 맞힌 화살 개수입니다. ( i는 0~10 사이의 정수입니다.)
    • 라이언이 가장 큰 점수 차이로 우승할 수 있는 방법이 여러 가지 일 경우, 가장 낮은 점수를 더 많이 맞힌 경우를 return 해주세요.
      • 가장 낮은 점수를 맞힌 개수가 같을 경우 계속해서 그다음으로 낮은 점수를 더 많이 맞힌 경우를 return 해주세요.
      • 예를 들어, [2,3,1,0,0,0,0,1,3,0,0]과 [2,1,0,2,0,0,0,2,3,0,0]를 비교하면 [2,1,0,2,0,0,0,2,3,0,0]를 return 해야 합니다.
      • 다른 예로, [0,0,2,3,4,1,0,0,0,0,0]과 [9,0,0,0,0,0,0,0,1,0,0]를 비교하면[9,0,0,0,0,0,0,0,1,0,0]를 return 해야 합니다.
  • 라이언이 우승할 방법이 없는 경우, return 할 정수 배열의 길이는 1입니다.
    • 라이언이 어떻게 화살을 쏘든 라이언의 점수가 어피치의 점수보다 낮거나 같으면 [-1]을 return 해야 합니다.

입출력 예

n / info / result

5 [2,1,1,1,0,0,0,0,0,0,0] [0,2,2,0,1,0,0,0,0,0,0]
1 [1,0,0,0,0,0,0,0,0,0,0] [-1]
9 [0,0,1,2,0,1,1,1,1,1,1] [1,1,2,0,1,2,2,0,0,0,0]
10 [0,0,0,0,0,0,0,0,3,4,3] [1,1,1,1,1,1,1,1,0,0,2]

제한시간 안내

  • 정확성 테스트 : 10초

# 문제 풀이

 

실제 이번 블라인드 코테에 참여하여서 3(1,2,3).5(6번 정확성) 문제를 풀었는데 이번 문제의 경우 마지막까지 잡고도 테스트 케이스 하나를 통과시키지 못해서 1차 코테(합격기준이 4 || 4.5솔)를 통과 못한 경험이 있다. 당시에는 노느라 준비를 덜했었기에 매우 아쉬웠는데 지금 와서 비교적 쉽게 풀어내는걸 보니 알고리즘 실력이 많이 늘었다는걸 체감했다.

문제를 풀면서 주의 할점이 몇가지 있는데 1.배열 길이 때문에 dfs나 완전 탐색으로 구해진 조합과 입력 받은 info의 점수를 비교해서 문제에서 요구하는 조건에 부합하는 조합들을 미리 필터링하면서 탐색해야한다는 것, 2.라이언이 이기지 못하는 경우가 꽤 다양하게 주어지니 [-1]를 얼리 리턴 하려고 하지 않는 것, 3. 같은 점수 차이의 경우 더 낮은 점수를 맞춘게 많은 것을 구하는 것 이 3가지 정도이다. 사실 이전 블라인드 코테에서는 dfs 같은 것에 익숙하지 않아서 해당 조합을 구하는 것 부터가 조금 어려웠다. 지금도 사실 그렇게 익숙한건 아니지만 비교적 간단한 응용은 어렵지 않아서 다행인거 같다. 

 

# 솔루션 플로우

1. 해당 솔루션의 진행 과정을 담을 result를 만들어준다.(gap: 0, answer [-1])

2. 입력 받은 n 과 info를 통해서 dfs를 진행하며 n을 모두 사용한 조합들을 구한다.

3. 구해진 조합과 입력 받은 info의 점수를 문제에서 요구하는 대로(0점 무시, 점수 약탈) 구한다.

4. 구해진 점수간의 차이가 저장된 gap보다 크거나 같을 경우 result의 answer와 gap을 변경해준다.

5. gap이 같을 경우 기존 result.answer와 구해진 조합 state를 비교해서 더 낮은 점수를 많이 맞춘 것을 구한다.

6. 모든 탐색이 끝난후 구해진 result.answer를 반환한다.

1. DFS

function solution(n, info) {
    const result = {
        gap: 0,
        answer: [-1]
    };

    const dfs = (n, index, stack = new Array(11).fill(0)) => {
        if (n === 0) {
            const state = Array.from(stack);
            const scores = checkScore(state, info);
            if (scores[0] > scores[1] && (scores[0] - scores[1]) >= result.gap) {
                if ((scores[0] - scores[1]) > result.gap) {
                    result.gap = (scores[0] - scores[1]);
                    result.answer = state;
                } else {
                    const [pre, current] = checkLevel(result.answer, state);
                    result.answer = pre >= current ?
                        result.answer :
                        state;
                }
            }
        }
        if (n > 0) {
            for (let idx = index; idx < info.length; idx++) {
                const state = n - (info[idx] + 1) >= 0 ?
                    info[idx] + 1 :
                    n;

                stack[idx] = state;
                dfs((n - state), (idx + 1), stack);
                stack[idx] = 0;
            };
        }
    }

    dfs(n, 0);
    
    return result.answer;

    function checkScore(ryan, ape) {
        return ryan.reduce((result, score, idx) => {
            if (score !== 0 || ape[idx] !== 0) {
                score <= ape[idx] ?
                    result[1] += (10 - idx) :
                    result[0] += (10 - idx)
            }
            return result;
        }, [0, 0]);
    }
    function checkLevel(arrA, arrB) {
        const pre = arrA.reduce((t, c, idx) => t + (c * (10 * idx)), 0);
        const current = arrB.reduce((t, c, idx) => t + (c * (10 * idx)), 0);

        return [pre, current];
    }
}
반응형