[프로그래머스 1] 수박수박수박수박수박수? (C/C++)

#include <string>
#include <vector>
using namespace std;
 
string solution(int n) 
{
    string answer = "";
    string plus = "수박";
    
    if(n == 1)
    {
        return "수";
    }
    else if(n == 2)
    {
        return "수박";
    }
    else
    {
        // 홀수
        if(n % 2 == 1)
        {
            for(int i = 1; i <= n/2; i++)
            {
                answer += plus;
            }
            answer += "수";
        }
        // 짝수
        else if(n % 2 == 0)
        {
            for(int i = 1; i <= n/2; i++)
            {
                answer += plus;
            }
        }
    }
 
    return answer;
}