[프로그래머스] 코딩테스트 연습-푸드파이트대회 (Java, C)

-링크

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

프로그래머스

 

 

-문제 설명

수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 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 이상인 경우만 입력으로 주어집니다.

 

-입출력 예

foodresult

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

 

 

 

-Java 해설

이 문제에서는 배열의 번호 = 음식의 종류 이기 때문에, 쉽게 문제 풀이에 접근이 가능하다.

배열 0번은 물에 해당하고, 제약사항에 보면 항상 1로 주어지기 때문에 0(물)을 기준으로 양쪽을 동일하게 배치하려고 한다면

배열 0번(물)은 언제나 2로 나눌 경우 몫이 0에 해당하기 때문에 따로 복잡한 예외처리가 필요없다.

C로 풀었다면 reverse를 직접 구현해야 해서 풀이가 더 길어지지만 JAVA에는 문자열을 reverse할 수 있는 함수가 이미 주어졌기 때문에 

비교적 간단히 풀이가 가능하다.

칼로리가 적은 순서대로 배치해야하는데, 칼로리가 적다 = 배열의 번호가 적다 이기때문에, 그냥 순서대로 for문을 돌리면 된다.

양쪽에 똑같은 양의 음식을 배치해야하기 때문에, 2로 나눌경우 나머지를 제외한 몫을 사용하면 된다 (4개 일경우 양쪽에 2개씩 배치 7개 일 경우 양쪽에 3개씩 동일하게 배치)

그 후 0을 기점으로 처음 구한 answer를 reverse 시켜서 붙이면 완성이다.

반드시 answer에 0을 붙이기 전 reverse시킬 StringBuffer에 담아 놓은 후 0을 붙여야 한다.

 

 

 

-풀이(Java)

class Solution {
    public String solution(int[] food) {
        String answer = "";
        
        
        for(int i = 0; i < food.length; i++){
            for(int j = 0; j < food[i]/2; j++){
                answer += Integer.toString(i);
            }
        }
        
        StringBuffer sb = new StringBuffer(answer);
        answer += "0";
        answer += sb.reverse().toString();
        
        return answer;
    }
}

 

 

 

-C해설

Java의 풀이와 비교해 봤을 때 가장 큰 차이점은 메모리 할당과 reverse함수를 직접 구현해야 한다는 점이다.

우선 메모리 할당을 할 때 크게 할당하는 것을 별로 안좋아해서 temp에 크게 할당 후 answer에는 알맞은 길이로 할당해서 return 하고 싶었지만 실제시험 시 시간을 고려하여 배열의 크기는 최대치로 할당해 주었다.

배열의 최대치는 배열의 길이(9번까지 제약사항이므로) * 한 배열당 나올수 있는 최대개수(1000개가 제약사항) 으로 구할 수 있다.

이렇기 때문에 제약사항을 잘 읽어 여러 테스트 사항에 에러가 나지 않도록 최대 사이즈로 구성해야한다.

여기서 malloc을 할 때 평소 실무에서도 최대 길이 +1을 꼭 해주는데, 그 이유는 문자열을 읽어드릴 때 NULL을 만날 때 까지 읽기 때문에

문자열의 길이를 가득 채울 때 맨 마지막에 들어 갈 NULL의 자리를 같이 할당 해 주는 것이다. ( 그렇지 않으면 최대 길이의 문자열이 테스트 케이스로 있을 때 반드시 코어가 발생(segmentation fault)한다. 

food의 배열은 int인데, return은 문자열로 해야하므로 간단하게 integer를 아스키코드로 변환하여 배열에 넣어주었다.

그 후0을 포함하지 않은 길이를 구해주고 0을 기준으로 배열을 거꾸로 읽어 문자열에 담는다.

맨 마지막에 길이를 구해 '\0'처리를 해 준 것은 쓰레기 값이 들어갈 것을 방지한 것이다.

위에 설명한 것 과 같이 문자열을 읽을 땐 NULL까지를 읽으므로 쓰레기 값이 들어 간 경우 쓰레기값까지 읽어들여 문제풀이에 실패하기 때문이다.

 

 

 

-풀이(C)

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>

// food_len은 배열 food의 길이입니다.
char* solution(int food[], size_t food_len) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    char* answer = (char*)malloc((food_len*1000)+1);
    int pos= 0, i = 0, len = 0, j = 0;
    
    memset(answer,0x00,sizeof(answer));
    
    for(i = 0; i < food_len; i++)
    {
        for(j = 0; j < food[i]/2; j++){
            answer[pos] = i+48;
            pos++;
        }
    }
    
    len = strlen(answer);
    answer[len] = '0';
    pos = 1;
    
    for(i = len -1; i >= 0; i--){
        answer[len+pos] = answer[i];
        pos++;
    }
    
    answer[len+pos]='\0';
    
    return answer;
}