[코딩테스트 Lv.1] 수박수박수박수박수박수?

문제 설명

길이가 n이고, "수박수박수박수...."와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 "수박수박"을 리턴하고 3이라면 "수박수"를 리턴하면 됩니다.

제한사항

n은 길이 10,000이하인 자연수입니다.

입출력 예

n	return
3	"수박수"
4	"수박수박"

---

내 풀이

Java

class Solution {
    public String solution(int n) {
        String answer = "";
        String[] sb = {"수", "박"};
        
        for(int i = 0; i < n; i++){
            int k = 1;
            if(i % 2 == 0) k = 0;
            answer += sb[k];
        }
        return answer;
    }
}

C

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

char* solution(int n) {
    // 리턴할 값은 메모리를 동적 할당해주세요.
    char* answer = (char*)malloc(sizeof(char) * 3 * n);
    
    strcpy(answer, "");
    int i = 0;
    for(i = 0;i < n; i++){
        if(i % 2 == 0){
            strcat(answer, "수");
        }else{
            strcat(answer, "박");
        }
    }
    
    
    return answer;
}

Python

def solution(n):
    answer = ''
    sb = ('수', '박')
    
    for i in range(n):        
        
        if i % 2 == 0:
            answer += sb[0]
        else:
            answer += sb[1]
        
    return answer

---

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

프로그래머스