[Level 1] 수박수박수박수박수박수? (C++/Python)

C++)

#include <string>
using namespace std;

string solution(int n) {
    string answer = "";
    
    for(int i = 0; i < n/2; i++)
        answer += "수박";
    
    answer += (n%2 == 1) ? "수" : "";
    
    return answer;
}

 

 

 

Python)

def solution(n):
    return "수박" * (n//2) + "수" * (n%2)