문제
수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 1대 1로 대결하며, 매 대결마다 음식의 종류와 양이 바뀝니다. 대결은 준비된 음식들을 일렬로 배치한 뒤, 한 선수는 제일 왼쪽에 있는 음식부터 오른쪽으로, 다른 선수는 제일 오른쪽에 있는 음식부터 왼쪽으로 순서대로 먹는 방식으로 진행됩니다. 중앙에는 물을 배치하고, 물을 먼저 먹는 선수가 승리하게 됩니다.이때, 대회의 공정성을 위해 두 선수가 먹는 음식의 종류와 양이 같아야 하며, 음식을 먹는 순서도 같아야 합니다. 또한, 이번 대회부터는 칼로리가 낮은 음식을 먼저 먹을 수 있게 배치하여 선수들이 음식을 더 잘 먹을 수 있게 하려고 합니다. 이번 대회를 위해 수웅이는 음식을 주문했는데, 대회의 조건을 고려하지 않고 음식을 주문하여 몇 개의 음식은 대회에 사용하지 못하게 되었습니다.
예를 들어, 3가지의 음식이 준비되어 있으며, 칼로리가 적은 순서대로 1번 음식을 3개, 2번 음식을 4개, 3번 음식을 6개 준비했으며, 물을 편의상 0번 음식이라고 칭한다면, 두 선수는 1번 음식 1개, 2번 음식 2개, 3번 음식 3개씩을 먹게 되므로 음식의 배치는 "1223330333221"이 됩니다. 따라서 1번 음식 1개는 대회에 사용하지 못합니다.
수웅이가 준비한 음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열 food가 주어졌을 때, 대회를 위한 음식의 배치를 나타내는 문자열을 return 하는 solution 함수를 완성해주세요.
입출력 예
food result [1, 3, 4, 6] "1223330333221" [1, 7, 1, 2] "111303111"
접근 방식
- 0번 인덱스는 무조건 물이니 제외한다.
- 1번 인덱스 값부터 2로 나눈 뒤 소수점을 제외한 만큼 정답 배열에 추가한다.
- 모든 배열의 값을 추가했다면 맨 뒤에 0을 추가한다.
- 정답 배열을 뒤집은 값을 0 뒤에 또 붙인다.
작성한 코드
function solution(food) { let answer = []; food.forEach((e, idx) => { if (idx === 0) { } else { const count = Math.floor(e / 2); answer = [...answer, ...Array(count).fill(idx)]; } }); return answer.join('') + '0' + answer.reverse().join(''); }
let answer = [];
정답을 저장할 배열.
food.forEach
순회하면서 기존 배열을 바탕으로 새 배열을 리턴할 건 아니고 answer 배열에 값만 추가할거라 forEach 사용
if (idx === 0)
물은 스킵하고
const count = Math.floor(e / 2);
해당 음식을 2로 나눈 뒤 홀수일 경우를 생각하여 소수점을 버림
answer = [...answer, ...Array(count).fill(idx)];
기존에 있던 값이 날아가지 않게 스프레드 연산자로 배열 유지하고
count 만큼의 길이를 idx 값으로 채우기
ex) Array(2).fill(3) = [기존값, 3, 3]
문제가 길어서 처음 봤을 땐 막막했는데, 막상 읽어보니 크게 어려운 문제는 아니었다.
배열을 원하는 길이만큼 원하는 값으로 채우는 방법이 뭐가 있을까 고민하다가 fill을 찾았는데, 생각해보니 repeat도 있었다.
function solution(food) {
let res = '';
for (let i = 1; i < food.length; i++) {
res += String(i).repeat(Math.floor(food[i]/2));
}
return res + '0' + [...res].reverse().join('');
}
이건 다른사람 풀이인데 나랑 다르게 for문을 사용했고, 배열이 아닌 문자열로 초기값을 세팅했다. 근데 어차피 물이 0번 인덱스라 제외할거였으면 for문을 사용하는 방법이 예외처리를 따로 안해도 되니까 더 편할 것 같다.
코테 문제는 뭔가... 접근 방식을 떠올리는 게 가장 어려운 것 같다.