프로그래머스 수박수박수박수박수박수? Swift

문제 설명

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

제한 조건

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

입출력 예

n	return
3	수박수
4	수박수박

풀이: 짝수일 땐 수박을 반복하고 홀수일 땐 수가 있는 상태에서 박수를 반복해서 더해준다.

func solution(_ n:Int) -> String {
    var str = String()
     var su = "수"
    if n == 1 {
        return "수"
    }
    else if n%2 == 0 {
        for _ in 1...n/2{
            str += "수박"
        }
        return str
    }else{
        for _ in 1...n/2{
            su += "박수"
        }
        return su
    }
}