[Level1] 푸드 파이트 대회 (C++/Python)

C++)

#include <string>
#include <vector>
 
using namespace std;
 
string solution(vector<int> food) 
{
    int ele = 0;
    string front = "";
    string back = "";
    
    for(int idx = 1; idx < food.size(); idx++)
    {
        ele = food.at(idx);
 
        for(int cnt = 0; cnt < ele/2; cnt++)
        {
            front += to_string(idx);
            back = to_string(idx) + back;
        }
    }
    
    return front+"0"+back;
}

 

 

 

Python)

def solution(food):
    answer = ''
    for i in range(1, len(food)):
        for _ in range(food[i]//2):
            answer += str(i)
    return answer + '0' + answer[::-1]