개발자 일기/알고리즘 공부

[프로그래머스 JS] 푸드 파이트 대회

MS_developer 2023. 8. 14. 21:48

📖 문제 설명

수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 1대 1로 대결하며, 매 대결마다 음식의 종류와 양이 바뀝니다. 대결은 준비된 음식들을 일렬로 배치한 뒤, 한 선수는 제일 왼쪽에 있는 음식부터 오른쪽으로, 다른 선수는 제일 오른쪽에 있는 음식부터 왼쪽으로 순서대로 먹는 방식으로 진행됩니다. 중앙에는 물을 배치하고, 물을 먼저 먹는 선수가 승리하게 됩니다.

 

이때, 대회의 공정성을 위해 두 선수가 먹는 음식의 종류와 양이 같아야 하며, 음식을 먹는 순서도 같아야 합니다. 또한, 이번 대회부터는 칼로리가 낮은 음식을 먼저 먹을 수 있게 배치하여 선수들이 음식을 더 잘 먹을 수 있게 하려고 합니다. 이번 대회를 위해 수웅이는 음식을 주문했는데, 대회의 조건을 고려하지 않고 음식을 주문하여 몇 개의 음식은 대회에 사용하지 못하게 되었습니다.

 

예를 들어, 3가지의 음식이 준비되어 있으며, 칼로리가 적은 순서대로 1번 음식을 3개, 2번 음식을 4개, 3번 음식을 6개 준비했으며, 물을 편의상 0번 음식이라고 칭한다면, 두 선수는 1번 음식 1개, 2번 음식 2개, 3번 음식 3개씩을 먹게 되므로 음식의 배치는 "1223330333221"이 됩니다. 따라서 1번 음식 1개는 대회에 사용하지 못합니다.

 

수웅이가 준비한 음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열 food 가 주어졌을 때, 대회를 위한 음식의 배치를 나타내는 문자열을 return 하는 solution 함수를 완성해주세요.


🚫 제한 사항

  • 2 ≤ food의 길이 ≤ 9
  • 1 ≤ food의 각 원소 ≤ 1,000
  • food에는 칼로리가 적은 순서대로 음식의 양이 담겨 있습니다.
  • food[i]는 i번 음식의 수입니다.
  • food[0]은 수웅이가 준비한 물의 양이며, 항상 1입니다.
  • 정답의 길이가 3 이상인 경우만 입력으로 주어집니다.

💾 입출력 예시

 

food result
[1, 3, 4, 6] "1223330333221"
[1, 7, 1, 2] "111303111"

 

 


⌨️ 나의 풀이 (코드)

 

먼저 의사 코드를 작성했다.

1. 반환값은 좌우가 대칭되는 수의 문자열이므로, food[0]을 제외한 나머지 수를 차례대로 기록, 기록하는 수는 i가 되며 음식의 번호(i)를 기록
2. 해당 음식을 짝수 개 (% 2)만큼 기록한다. 이때, 0을 기준으로 좌우에 배치한다. => 해당 개수만큼 unshift 와 push를 사용해야하므로 배열의 형태로 기록
3. 음식의 길이만큼 반복, 완성된 배열을 차례대로 문자열로 구성한다.

 

위 내용을 기반으로 코드를 작성했다.

 

function solution(food) {
  var answer = "";
  const stringArr = [];
  for (let i = 0; i < food.length; i++) {
    if (i === 0) {
      stringArr.push(i);
    } else {
      console.log(food[i]);
      const itrNum = Math.floor(food[i] / 2);
      for (let j = 0; j < itrNum; j++) {
        stringArr.push(i);
        stringArr.unshift(i)
      }
    }
  }

  console.log(stringArr);
  return answer;
}

 

코드를 구현하는 과정에서 이상한 점을 느꼈다.

 

위의 코드는 순서가 반대로 되어 있었다.

 

생각의 흐름 상 첫 번째 요소인 물을 먼저 배치한 후 해당 배열의 앞뒤로 요소를 추가하는 방식이 자연스러워 보였으나, 해당 로직을 사용하면 최종 과정 전에 사용되는 배열의 요소들이 0은 올바른 위치에 있으나 나머지 요소들의 순서가 반대로 되어 있었다. (i.e. [ 1, 2, 2, 3, 3, 3, 0, 3, 3, 3, 2, 2, 1 ] => [ 3, 3, 3, 2, 2, 1, 0, 1, 2, 2, 3, 3, 3 ] )

 

이를 기반으로 코드의 접근 방식을 수정해 보았다.

 

1. 반환값은 좌우가 대칭되는 수의 문자열이므로, food[0]을 제외한 나머지 수를 차례대로 기록, 기록하는 수는 i가 되며 음식의 번호(i)를 기록
2. 해당 음식을 짝수 개 (% 2)만큼 기록한다. 이때, 주어진 배열 food의 길이만큼의 배열을 반복하되, 마지막 요소(i = food.length; i >= 0; i--)에서부터 시작해 해당 개수만큼 unshift 와 push를 사용해 배열을 기록한다.
3. 완성된 배열을 차례대로 문자열로 구성한다.

 

즉, 배열의 순회 방식을 처음(0)부터 시작하는 방법에서 배열의 끝에서부터 시작하는 방식으로 수정했다.

 

function solution(food) {
  const stringArr = [];
  for (let i = food.length; i >= 0; i--) {
    if (i === 0) {
      stringArr.splice(stringArr.length / 2, 0, 0);
    } else {
      const itrNum = Math.floor(food[i] / 2);
      for (let j = 0; j < itrNum; j++) {
        stringArr.push(i);
        stringArr.unshift(i);
      }
    }
  }

  return stringArr.join("");
}

 

이후 모든 테스트 케이스를 통과했다.

 


📝 Note

 

코드를 구현하면서 문득 "굳이 배열로 만들고 (push, unshift) 이후에 문자열로 다시 정리하는 과정이 필요한가?"라는 생각이 들었다.

 

다른 사람의 코드에서 배열로 만드는 과정을 줄일 수 있는 방법을 알 수 있었다.(탈퇴한 사용자님의 코드)

 

function solution(food) {
  let res = "";
  for (let i = 1; i < food.length; i++) {
    res += String(i).repeat(Math.floor(food[i] / 2));
    console.log(res)
  }

  return res + "0" + [...res].reverse().join("");
}

 

먼저 반복문을 통해 배열의 앞 부분을 만든다. 위 과정에서 .repeat() 메서드를 활용했고, 이후 해당 배열의 마지막에 문자열 "0"을 더한 후 거기에 기존 문자열을 .reverse() 메서드로 반전시킨 뒤 추가하는 방식이었다.