[Algorithm] 양궁대회 ( Programmers > 2022 KAKAO BLIND RECRUTMENT - Python, JavaScript )

문제 풀이

카카오배 양궁대회가 열렸습니다.

라이언은 저번 카카오배 양궁대회 우승자이고 이번 대회에도 결승전까지 올라왔습니다. 결승전 상대는 어피치입니다.

카카오배 양궁대회 운영위원회는 한 선수의 연속 우승보다는 다양한 선수들이 양궁대회에서 우승하기를 원합니다. 따라서, 양궁대회 운영위원회는 결승전 규칙을 전 대회 우승자인 라이언에게 불리하게 다음과 같이 정했습니다.

 

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

어피치와 라이언이 아래와 같이 화살을 맞힐 경우,

 

어피치의 최종 점수는 15점, 라이언의 최종 점수는 19점입니다. 4점 차이로 라이언이 우승합니다.

 

하지만, 라이언이 아래와 같이 화살을 맞힐 경우 더 큰 점수 차로 우승할 수 있습니다.

 

어피치의 최종 점수는 17점, 라이언의 최종 점수는 23점입니다. 6점 차이로 라이언이 우승합니다.

따라서 [0, 2, 2, 0, 1, 0, 0, 0, 0, 0, 0] 을 return 해야 합니다.

 

 

입출력 예 #2

라이언이 10점을 맞혀도 어피치가 10점을 가져가게 됩니다.

따라서, 라이언은 우승할 수 없기 때문에 [-1] 을 return 해야 합니다.

 

 

입출력 예 #3

어피치와 라이언이 아래와 같이 화살을 맞힐 경우,

 

어피치의 최종 점수는 10점, 라이언의 최종 점수는 39점입니다. 29점 차이로 라이언이 우승합니다.

 

하지만 라이언이 아래와 같이 화살을 맞힐 경우, 

 

어치피의 최종 점수는 13점, 라이언의 최종 점수는 42점입니다. 이 경우도 29점 차이로 라이언이 우승합니다.

하지만, 첫 번째 경우와 두 번째 경우를 비교했을 때, 두 번째 경우가 두 경우 중 가장 낮은 점수인 4점을 더 많이 맞혔기 때문에 [1, 1, 2, 3, 0, 2, 0, 0, 0, 0, 0] 이 아닌 [1, 1, 2, 0, 1, 2, 2, 0, 0, 0, 0] 을 return 해야 합니다.

 

 

입출력 예 #4

가장 큰 점수 차이로 이기는 경우 중에서 가장 낮은 점수를 가장 많이 맞힌, 10~3점을 한 발씩 맞히고 나머지 두 발을 0점에 맞히는 경우인 [1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 2] 를 return 해야 합니다.

 

 

 

---

 

 

 

문제 풀이

양궁대회 알고리즘 문제는 큐를 사용하여 문제를 해결하였다. 기본적인 문제 풀이의 해결 방법은 큐를 while문을 사용하여 계속 queue에다가 추가해주다가 점수 == n 같거나 queue에서 나온 배열의 길이가 10과 같을때까지 추가해주고, 전과 같은 조건이 해당이 된다면 점수를 비교하여 answer 배열의 값을 변경해주는 방식이다.

 

python:

def solution(n, info):
    answer = []
 
    # 어피치 총합 점수
    apeach = sum([10 - i for i in range(10) if info[i]])
 
    # 실질 획득 점수
    score = [(10 - i) * 2 if info[i] else 10 - i for i in range(10)]
 
    # 10 점을 안쏘는 기본값 설정
    queue = [[0]]
 
    # 10 점을 쏴서 점수를 얻을 수 있다면 추가
    if n >= info[0] + 1:
        queue.append([info[0] + 1])
 
    while queue:
        score_list = queue.pop(0)
 
        # 점수가 n과 같거나 score_list 길이가 10일 경우
        if sum(score_list) == n or len(score_list) == 10:
            original_score = sum([score[i] for i in range(len(answer)) if answer[i]])
            new_score = sum([score[i] for i in range(len(score_list)) if score_list[i]])
 
            if new_score > apeach and new_score >= original_score:
                answer = score_list
 
        # 쏠 수 있는 화살이 남아 있을 경우 어피치 보다 + 1 쏴서 점수를 획득할 수 있는 경우
        elif n >= sum(score_list) + info[len(score_list)] + 1:
            queue.append(score_list + [info[len(score_list)] + 1])
            queue.append(score_list + [0])
 
        # 더 이상 쏠 수 있는 화살이 없을 경우
        else:
            queue.append(score_list + [0])
 
    # 어치피 보다 높은 점수를 받을 수 없을때
    if not answer:
        return [-1]
 
    # 쏜 점수 + 안 쏜곳은 0점 처리 + 0점 자리에 남은거 쏘기
    return answer + [0] * (10 - len(answer)) + [n - sum(answer)]

 

 

 

 

javascript:

function solution(n, info) {
  let answer = [];
 
  let apeach = 0;
  const score = info.map((score, index) => {
    if (score) apeach += 10 - index;
 
    return score ? (10 - index) * 2 : 10 - index;
  });
 
  const queue = [[0]];
  if (n >= info[0] + 1) queue.push([info[0] + 1]);
 
  while (queue.length) {
    const scoreList = queue.shift();
    const scoreSum = scoreList.reduce(
      (accumulator, currentValue) => accumulator + currentValue,
      0
    );
 
    if (scoreSum === n || scoreList.length === 10) {
      let originalScore = 0;
      for (let i = 0; i < answer.length; ++i) {
        if (answer[i]) originalScore += score[i];
      }
 
      let newScore = 0;
      for (let i = 0; i < scoreList.length; ++i) {
        if (scoreList[i]) newScore += score[i];
      }
 
      if (newScore > apeach && newScore >= originalScore) {
        answer = scoreList;
      }
    } else if (n >= scoreSum + (info[scoreList.length] + 1)) {
      queue.push([...scoreList, info[scoreList.length] + 1]);
      queue.push([...scoreList, 0]);
    } else {
      queue.push([...scoreList, 0]);
    }
  }
 
  if (!answer.length) return [-1];
 
  return [
    ...answer,
    ...Array.from({ length: 10 - answer.length }, () => 0),
    n - answer.length,
  ];
}

 

 

 

 

 

 

 

 

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

프로그래머스